dgl.ops.copy_e_mean

dgl.ops.copy_e_mean(g, x)

广义 SpMM 函数。它将边特征视为消息。然后通过在目标节点上求平均来聚合消息。

参数:

• g (DGLGraph) – 输入图

• x (tensor) – 边特征。

返回:

结果张量。

返回类型:

张量

说明

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

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