dgl.ops.u_dot_e

dgl.ops.u_dot_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。