dgl.NodeFlow.prop_flow¶
-
NodeFlow.
prop_flow
(message_funcs='default', reduce_funcs='default', apply_node_funcs='default', flow_range='__ALL__', inplace=False)[source]¶ Perform the computation on flows. By default, it runs on all blocks, one-by-one. On block i, it runs pull on nodes in layer i+1, which generates messages on edges in block i, runs the reduce function and node update function on nodes in layer i+1.
Users can specify a list of message functions, reduce functions and node apply functions, one for each block. Thus, when a list is given, the length of the list should be the same as the number of blocks.
- Parameters
message_funcs (a callable, a list of callable, optional) – Message functions on the edges. The function should be an
Edge UDF
.reduce_funcs (a callable, a list of callable, optional) – Reduce functions on the node. The function should be a
Node UDF
.apply_node_funcs (a callable, a list of callable, optional) – Apply functions on the nodes. The function should be a
Node UDF
.flow_range (int or a slice or ALL.) – The specified blocks to run the computation.
inplace (bool, optional) – If True, update will be done in place, but autograd will break.