入门
高级材料
API 参考
min()
说明
杂项
内置规约函数,通过取最小值来聚合消息。
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)}