dgl.ops.copy_e_sum

dgl.ops.copy_e_sum(g, x)

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

Parameters:
  • g (DGLGraph) – The input graph

  • x (tensor) – The edge 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.