dgl.topological_nodes_generator

dgl.topological_nodes_generator(graph, reverse=False)[源代码]

使用拓扑遍历的节点前沿生成器。

参数:

graph (DGLGraph) – 图对象。
reverse (bool, 可选) – 如果为 True，则沿着入边的方向遍历。

返回值:

每个节点前沿是节点 ID 的列表或张量。

返回类型:

list of node frontiers

示例

给定一个图（有向，边从较小节点 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.topological_nodes_generator(g))
[tensor([0]), tensor([1]), tensor([2]), tensor([3, 4]), tensor([5])]