dgl.function.min

dgl.function.min(msg, out)

内置的reduce函数,用于通过最小值聚合消息。

Parameters:
  • msg (str) – The message field.

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

示例

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