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)

使用标记 DFS 生成的边前沿进行消息传播。

参数:

• graph (DGLGraph) – 图对象。

• source (list, tensor of nodes) – 源节点。

• message_func (callable, optional) – 消息函数。

• reduce_func (callable, optional) – 归约函数。

• reverse (bool, optional) – 如果为 True，则沿入边方向遍历。

• has_reverse_edge (bool, optional) – 如果为 True，则包含反向边。

• has_nontree_edge (bool, optional) – 如果为 True，则包含非树边。

• apply_node_func (callable, optional) – 更新函数。

另请参阅

dgl.traversal.dfs_labeled_edges_generator