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.