PyG (PyTorch Geometric)，基于 PyTorch 编写和训练图神经网络 (GNN)
- 集成了各种针对图或其他不规则结构的深度学习方法
- 包括易用小批量加载器，可在许多小型和单个巨型图上运行
- 多 GPU 支持，torch.compile 支持，DataPipe 支持
- 内置大量基准数据集、支持神经网络的
发表评论
2181 views