torch_geometric.data

数据对象

Data	描述同构图的数据对象。
HeteroData	描述异构图的数据对象，在分离的存储对象中保存多种节点和/或边类型。
Batch	描述一批图作为一个大（不连通）图的数据对象。
TemporalData	由描述时间图的事件流组成的数据对象。
数据集	用于创建图数据集的Dataset基类。
InMemoryDataset	用于创建易于适应CPU内存的图数据集的Dataset基类。
OnDiskDataset	用于创建大型图数据集的基类，这些数据集不容易一次性放入CPU内存，通过利用Database后端进行磁盘存储和数据对象的访问。
LargeGraphIndexer	对于一个由多个子图组成的数据集，假设这些子图是一个更大图的一部分，将这些值整理到一个大型图存储中以节省资源。
TripletLike	alias of Tuple[str, str, str]

远程后端接口

FeatureStore	一个抽象基类，用于从远程特征存储中访问特征。
GraphStore	一个抽象基类，用于从远程图存储访问边。
TensorAttr	定义FeatureStore张量的属性。
EdgeAttr	定义GraphStore边的属性。

数据库

Database	用于从数据库中插入和检索数据的基类。
SQLiteDatabase	An index-based key/value database based on sqlite3.
RocksDatabase	An index-based key/value database based on RocksDB.

PyTorch Lightning 包装器

LightningDataset	将一组Dataset对象转换为pytorch_lightning.LightningDataModule变体。
LightningNodeData	将Data或HeteroData对象转换为pytorch_lightning.LightningDataModule变体。
LightningLinkData	将Data或HeteroData对象转换为pytorch_lightning.LightningDataModule变体。

辅助函数

makedirs	递归地创建一个目录。
download_url	将URL的内容下载到特定文件夹。
download_google_url	将Google Drive ID的内容下载到特定文件夹。
extract_tar	将tar存档提取到特定文件夹。
extract_zip	将zip存档提取到特定文件夹。
extract_bz2	将bz2压缩包提取到特定文件夹。
extract_gz	将gz存档提取到特定文件夹。
get_features_for_triplets	对于给定的一组三元组，检索包含的Data对象
get_features_for_triplets_groups	给定一个索引器和一系列三元组（如数据集），从索引中检索每个三元组的指定节点和边特征。