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.