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`.