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 (DGLHeteroGraph) – 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.