AI工具

pytorch_geometric

pyg-team/pytorch_geometric

基于PyTorch的图神经网络库,提供GNN模型、数据加载与基准数据集,支持图、网格与点云上的深度学习。

标签

  • deep-learning
  • geometric-deep-learning
  • graph-neural-networks
  • pytorch