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