dgl.ops.copy_u_mean

dgl.ops.copy_u_mean(g, x)

Generalized SpMM function. It copies node feature to edge as the message. Then aggregates the message by mean 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 mean function will return zero for nodes with no incoming messages.