dgl.ops.e_div_u

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

广义 SDDMM 函数。它通过将边特征除以源节点特征来计算边特征。

参数
  • g (DGLGraph) – 输入图

  • x (张量) – 边特征。

  • y (张量) – 源节点特征。

返回值

结果张量。

返回类型

张量

备注

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

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