dgl.ops.copy_u

dgl.ops.copy_u(g, x)[source]

广义SDDMM函数,将源节点特征复制到边。

Parameters:
  • g (DGLGraph) – The input graph.

  • x (tensor) – The source node features.

Returns:

结果张量。

Return type:

张量

注释

此函数支持自动梯度(根据输出梯度计算输入梯度)。