@torch-geometric
PyTorch Geometric 是一个基于 PyTorch 构建的库,用于开发和训练图神经网络 (GNN)。将此技能应用于图形和不规则结构的深度学习,包括小批量处理、多 GPU 训练和几何深度学习应用。