copy_u_max

copy_u_max(rowptr, colind, node_feat)

Generalized SpMM function. It copies node feature to edge as the message. Then aggregates the message by max on destination nodes.

Parameters:

• rowptr -row pointer in CSR format

• colind -col indices of graph

• node_feat -the source node features

Returns: The result tensor

Return type: tensor