pytorch_geometric:PyTorch的几何深度学习扩展库 源码

| | | | PyTorch几何(PYG)是几何深度学习扩展库 。 它包括从各种已发表的论文中对图形和其他不规则结构进行深度学习的各种方法,也称为。 此外,它包括一个易于使用的迷你批处理程序,可用于许多小的和单个巨型图,多GPU支持,大量通用基准数据集(基于创建自己的简单界面)以及有用的转换,既可以在任意图形上学习,也可以在3D网格或点云上学习。 PyTorch Geometric使实现Graph Neural Networks变得轻而易举(请参阅 的随附教程)。 例如,这是实现: import torch from torch . nn import Sequential as S