dgl.function.mean

dgl.function.mean(msg, out)

内置的reduce函数，通过平均值聚合消息。

Parameters:

• msg (str) – The message field.

• out (str) – The output node feature field.

示例

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