dgl.ops.copy_u_max

dgl.ops.copy_u_max(g, x)

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

Parameters:

• g (DGLGraph) – The input graph

• x (tensor) – The source node features.

Returns:

结果张量。

Return type:

张量

注释

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

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