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