dgl.DGLGraph.register_apply_node_func

DGLGraph.register_apply_node_func(func)[source]

Register global node apply function.

Once registered, func will be used as the default apply node function. Related operations include apply_nodes(), recv(), send_and_recv(), push(), pull(), update_all().

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