pytorch_geometric

PyTorch Geometric 是一个基于 PyTorch 的图神经网络库，用于处理结构化数据。它提供了多种方法来实现深度学习在图和其他不规则结构上的应用，涵盖了从已发表论文中提取的各种几何深度学习技术。核心功能包括易于使用的迷你批处理加载器、多GPU支持、`torch.compile` 支持以及 `DataPipe` 支持等，还内置了大量常用基准数据集和转换工具，适用于任意图、3D网格或点云的学习任务。该项目适合需要对图结构数据进行建模与分析的场景，如社交网络分析、分子结构预测等领域。