dgl.ops.u_div_e_sum

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

广义SpMM函数。 它将两个步骤融合到一个内核中。

1. 通过源节点和边特征计算消息。

2. 将消息按目标节点上的特征进行求和聚合。

Parameters:

• g (DGLGraph) – 输入图

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

• y (tensor) – 边缘特征。

Returns:

结果张量。

Return type:

张量

注释

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

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

对于没有传入消息的节点，sum函数将返回零。