torch_geometric.utils.subgraph

`torch_geometric.utils.subgraph`是PyTorch Geometric中的一个函数，用于从一个大的图中提取子图。

该函数的输入参数为原始图`data`和待提取的节点索引列表`sub_idx`，返回一个包含子图中节点和边的新图对象`sub_data`。此外，还可以指定是否将边的属性也复制到新图中。

使用示例：

import torch_geometric.utils as utils

# 原始图数据对象
data = ...

# 从原始图中提取节点索引为[1, 2, 3]的子图
sub_idx = [1, 2, 3]
sub_data = utils.subgraph(data, sub_idx)

# 从原始图中提取节点索引为[1, 2, 3]的子图，并将边的属性也复制到新图中
sub_data_with_edge_attr = utils.subgraph(data, sub_idx, relabel_nodes=True)