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.
- message_funcs (a callable, a list of callable, optional) – Message functions on the edges. The function should be
an