dgl.prop_edges_dfsο
- dgl.prop_edges_dfs(graph, source, message_func, reduce_func, reverse=False, has_reverse_edge=False, has_nontree_edge=False, apply_node_func=None)[source]ο
Message propagation using edge frontiers generated by labeled DFS.
- Parameters:
graph (DGLGraph) β The graph object.
source (list, tensor of nodes) β Source nodes.
message_func (callable, optional) β The message function.
reduce_func (callable, optional) β The reduce function.
reverse (bool, optional) β If true, traverse following the in-edge direction.
has_reverse_edge (bool, optional) β If true, REVERSE edges are included.
has_nontree_edge (bool, optional) β If true, NONTREE edges are included.
apply_node_func (callable, optional) β The update function.
See also
dgl.traversal.dfs_labeled_edges_generator