dgl.propagate.prop_edges_dfs

dgl.propagate.prop_edges_dfs(graph, source, reverse=False, has_reverse_edge=False, has_nontree_edge=False, message_func='default', reduce_func='default', apply_node_func='default')

Message propagation using edge frontiers generated by labeled DFS.

Parameters:

• graph (DGLGraph) – The graph object.
• source (list, tensor of nodes) – Source nodes.
• reverse (bool, optional) – If true, traverse following the in-edge direction.
• message_func (callable, optional) – The message function.
• reduce_func (callable, optional) – The reduce function.
• apply_node_func (callable, optional) – The update function.

See also

dgl.traversal.dfs_labeled_edges_generator()