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]

使用由标记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, 可选) – 如果为真，则包含反向边。

  • has_nontree_edge (bool, optional) – 如果为真，则包含NONTREE边。

  • apply_node_func (callable, optional) – The update function.

另请参阅

dgl.traversal.dfs_labeled_edges_generator