dgl.ops.copy_u
- dgl.ops.copy_u(g, x)[source]
Generalized SDDMM function that copies source node features to edges.
- 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).