dgl.function.max

dgl.function.max(msg, out)

内建的聚合函数，通过取最大值来聚合消息。

参数:

• msg (str) – 消息字段。

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

示例

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

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

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