dgl.ops.u_div_e_max

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

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

  1. 通过将源节点特征除以边特征来计算消息。

  2. 通过取最大值来聚合消息，作为目标节点的特征。

参数:

  • g (DGLGraph) – 输入图

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

  • y (tensor) – 边特征。

返回值:

结果张量。

返回类型:

张量

说明

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

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

对于没有传入消息的节点，max 函数将返回零。这是通过将所有负无穷大值替换为零来实现的。