dgl.DGLGraph.register_reduce_func

DGLGraph.register_reduce_func(func)[source]

Register global message reduce function.

Once registered, func will be used as the default message reduce function in message passing operations, including recv(), send_and_recv(), push(), pull(), update_all().

Parameters:func (callable) – Reduce function on the node. The function should be a Node UDF.