dgl.ops.copy_u_max

dgl.ops.copy_u_max(g, x)

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

参数:

  • g (DGLGraph) – 输入图

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

返回:

结果张量。

返回类型:

张量

注意事项

此函数支持自动求导（根据输出梯度计算输入梯度）。

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