dgl.NodeFlow.register_apply_node_func¶

NodeFlow.register_apply_node_func(func, block_id='__ALL__')[source]¶

Register global node apply function for a block.

Once registered, func will be used as the default apply node function. Related operations include apply_layer(), block_compute(), prop_flow().

Parameters:	func (callable) – Apply function on the nodes. The function should be a `Node UDF`. block_id (int or ALL) – the block to register the apply node function.