dgl.ops.copy_u_mean

dgl.ops.copy_u_mean(g, x)

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

参数：

• g (DGLGraph) – 输入图

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

返回值：

结果张量。

返回类型：

张量

注意

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

平均函数对于没有传入消息的节点将返回零。