PyTorch Geometric

PyTorch Geometric

PyTorch Geometric(PyG)是一个基于PyTorch的几何深度学习扩展库,用于处理图、点云等不规则结构数据。 该库由德国多特蒙德工业大学的研究者Matthias Fey和Jan E. Lenssen开发。它提供了丰富的图神经网络层实现、专用的CUDA内核以及易用的消息传递API。

PyG发布了2.0版本,提供了全面的异构图支持和GraphGym实验管理平台集成。 从2.5版本开始,其图神经网络层实现了与TorchScript的完全兼容。

想要了解更多“PyTorch Geometric”的信息,请点击:PyTorch Geometric百科