dgl.bfs_edges_generator

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

使用广度优先搜索生成边前沿。

参数：

  • graph (DGLGraph) – 图对象。

  • source (list, 节点张量) – 源节点。

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

返回值：

每个边前沿是一个边 ID 列表或张量。

返回值类型：

边前沿的list

示例

给定一个图（有向图，边从小的节点 ID 指向大的节点 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_edges_generator(g, 0))
[tensor([0]), tensor([1, 2]), tensor([4, 5])]