PyTorch几何

PyTorch几何是一个几何深度学习扩展库PyTorch．几何深度学习的目标是建立可以从非欧几里得数据(如图表)中学习的神经网络。基于图网络的机器学习方法在社交网络建模和生物医学领域，特别是在药物发现方面越来越受到关注。PyTorch geometry提供了一个易于使用的库，用于设计复杂的图网络问题，如蛋白质结构表示。它支持GPU和CPU，并基于最近的研究，包括一个很好的基于图的机器学习算法集合。