dgl.ops.copy_u_min

dgl.ops.copy_u_min(g, x)

广义 SpMM 函数。它将节点特征作为消息复制到边上。然后通过对目标节点上的消息取最小值进行聚合。

参数:

  • g (DGLGraph) – 输入图

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

返回值:

结果张量。

返回类型:

tensor

注意事项

此函数支持自动微分 (autograd) (给定输出梯度，计算输入梯度)。

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