dgl.slice_batch

dgl.slice_batch(g, gid, store_ids=False)[source]

从一批图中获取特定的图。

Parameters:

g (DGLGraph) – 输入的批量图。
gid (int) – 要检索的图的ID。
store_ids (bool) – 如果为 True，它将在生成的图的 ndata 和 edata 中分别以 dgl.NID 和 dgl.EID 的名称存储提取的节点和边的原始 ID。

Returns:

检索到的图表。

Return type:

DGLGraph

示例

以下示例使用PyTorch后端。

>>> import dgl
>>> import torch

创建一个批处理图。

>>> g1 = dgl.graph(([0, 1], [2, 3]))
>>> g2 = dgl.graph(([1], [2]))
>>> bg = dgl.batch([g1, g2])

获取第二个组件图。

>>> g = dgl.slice_batch(bg, 1)
>>> print(g)
Graph(num_nodes=3, num_edges=1,
      ndata_schemes={}
      edata_schemes={})