dgl.function.min

dgl.function.min(msg, out)

内置规约函数,通过取最小值来聚合消息。

参数:
  • msg (str) – 消息字段。

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

示例

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

上述示例等效于以下用户自定义函数(如果使用 PyTorch)

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