dgl.bfs_nodes_generator

dgl.bfs_nodes_generator(graph, source, reverse=False)[source]

使用广度优先搜索的节点边界生成器。

Parameters:

graph (DGLGraph) – The graph object.
source (list, tensor of nodes) – Source nodes.
reverse (bool, default False) – If True, traverse following the in-edge direction.

Returns:

每个节点边界是节点ID的列表或张量。

Return type:

list 节点边界列表

示例

给定一个图（有向，边从小节点ID到大节点ID）：

      2 - 4
     / \
0 - 1 - 3 - 5

>>> g = dgl.graph(([0, 1, 1, 2, 2, 3], [1, 2, 3, 3, 4, 5]))
>>> list(dgl.bfs_nodes_generator(g, 0))
[tensor([0]), tensor([1]), tensor([2, 3]), tensor([4, 5])]