dgl.ops.copy_u

dgl.ops.copy_u(g, x)[源码]

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

参数:
  • g (DGLGraph) – 输入图。

  • x (tensor) – 源节点特征。

返回值:

结果张量。

返回值类型:

张量

注意事项

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