dgl.ops.u_sub_e_min
- dgl.ops.u_sub_e_min(g, x, y)
广义 SpMM 函数。它将两个步骤融合到一个核中。
通过源节点特征和边特征的相减计算消息。
通过取最小值聚合消息作为目标节点上的特征。
- 参数：
g (DGLGraph) – 输入图
x (tensor) – 源节点特征。
y (tensor) – 边特征。
- 返回值：
结果张量。
- 返回值类型：
张量
说明
此函数支持自动微分（给定输出梯度计算输入梯度）。如果两个输入操作数的特征形状不匹配，我们首先将特征广播到统一的形状（注意内存使用不会相应增加），然后执行操作。
广播遵循 NumPy 语义。请参阅 https://docs.scipy.org.cn/doc/numpy/user/basics.broadcasting.html 获取有关 NumPy 广播语义的更多详细信息。
对于没有传入消息的节点，min 函数将返回零。这是通过将所有无穷大值替换为零来实现的。