dgl.function.mean

dgl.function.mean(msg, out)

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

参数:

• msg (str) – 消息字段。

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

示例

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

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

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