dgl.ops.v_add_e

dgl.ops.v_add_e(g, x, y)

广义 SDDMM 函数。它通过添加目标节点特征和边特征来计算边特征。

参数：

  • g (DGLGraph) – 输入图

  • x (tensor) – 目标节点特征。

  • y (tensor) – 边特征。

返回：

结果张量。

返回类型：

tensor

注意事项

此函数支持自动求导（根据输出梯度计算输入梯度）。如果两个输入操作数的特征形状不匹配，我们首先将特征广播到统一形状（请注意内存使用不会相应增加），然后执行操作。

广播遵循 NumPy 语义。有关 NumPy 广播语义的更多详细信息，请参见 https://docs.scipy.org.cn/doc/numpy/user/basics.broadcasting.html。