dgl.DGLGraph.register_message_func¶

DGLGraph.register_message_func(func)[source]¶

Register global message function.

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

Parameters:	func (callable) – Message function on the edge. The function should be an `Edge UDF`.