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).