Get Started
Advanced Materials
API Reference
Notes
Misc
dgl.function.
mean
Builtin reduce function that aggregates messages by mean.
msg (str) – The message field.
out (str) – The output node feature field.
Examples
>>> import dgl >>> reduce_func = dgl.function.mean('m', 'h')
The above example is equivalent to the following user defined function (if using PyTorch):
>>> import torch >>> def reduce_func(nodes): >>> return {'h': torch.mean(nodes.mailbox['m'], dim=1)}