dgl.function.sum

dgl.function.sum(msg, out)

内置的归约函数，通过求和聚合消息。

参数:

• msg (str) – 消息字段。

• out (str) – 输出节点特征字段。

示例

>>> import dgl
>>> reduce_func = dgl.function.sum('m', 'h')

上述示例等价于以下用户自定义函数（如果使用 PyTorch）

>>> import torch
>>> def reduce_func(nodes):
>>>     return {'h': torch.sum(nodes.mailbox['m'], dim=1)}