dgl.ops.copy_e_min

dgl.ops.copy_e_min(g, x)

广义 SpMM 函数。将边特征视为消息。然后通过在目标节点上取最小值来聚合消息。

参数:
  • g (DGLGraph) – 输入图

  • x (tensor) – 边特征。

返回:

结果张量。

返回类型:

张量

注意事项

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

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