dgl.ops.copy_u_sum

dgl.ops.copy_u_sum(g, x)

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

Parameters:
  • g (DGLGraph) – The input graph

  • x (tensor) – The source node features.

Returns:

The result tensor.

Return type:

tensor

Notes

This function supports autograd (computing input gradients given the output gradient).

The sum function will return zero for nodes with no incoming messages.