dgl.propagate.prop_edges_dfs

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

Message propagation using edge frontiers generated by labeled DFS.

Parameters:
  • graph (DGLGraph) – The graph object.
  • source (list, tensor of nodes) – Source nodes.
  • reversed (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.