dgl.NodeFlow.register_apply_edge_func¶

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

Register global edge apply function for a block.

Once registered, func will be used as the default apply edge function in apply_block().

Parameters

func (callable) – Apply function on the edge. The function should be an Edge UDF.
block_id (int or ALL) – the block to register the apply edge function.