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