dgl.function.min
- dgl.function.min(msg, out)
内置的reduce函数,用于通过最小值聚合消息。
示例
>>> 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)}