dgl.function.sum

dgl.function.sum(msg, out)

内置的reduce函数，通过求和来聚合消息。

Parameters:

• 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)}