ICLRMar, 2019

使用 PyTorch Geometric 进行快速的图形表征学习

TL;DRPyTorch Geometric 是一个基于 PyTorch 的深度学习库,用于处理不规则结构的输入数据,如图形,点云和流形。该库通过利用稀疏 GPU 加速,提供专用的 CUDA 内核以及引入对不同大小的输入示例的高效小批量处理,实现了高数据吞吐量。本文详细介绍了该库,并在同质评估场景下对实现的方法进行了全面比较研究。