PyG 文档

PyG (PyTorch Geometric) 是一个基于 PyTorch 的库，用于轻松编写和训练图神经网络（GNNs），适用于与结构化数据相关的广泛应用。

它包含了从各种已发表的论文中提取的用于图和其他不规则结构深度学习的多种方法，也被称为几何深度学习。此外，它还包括易于使用的小批量加载器，用于操作许多小型和单个巨型图，多GPU支持，torch.compile 支持，DataPipe 支持，大量常见的基准数据集（基于简单的接口创建您自己的数据集），以及有用的转换，既适用于任意图的学习，也适用于3D网格或点云。