基于 PyTorch Geometric 的快速图表示学习
机器学习
2019-04-26 v3 机器学习
摘要
我们介绍 PyTorch Geometric，一个构建于 PyTorch 之上、用于图、点云和流形等不规则结构输入数据深度学习的库。除通用图数据结构与处理方法外，它还包含关系学习和 3D 数据处理领域多种近期发表的方法。PyTorch Geometric 通过利用稀疏 GPU 加速、提供专用 CUDA 内核以及引入针对不同大小输入样本的高效小批量处理，实现了高数据吞吐率。在本文中，我们详细介绍了该库，并在同质评估场景下对所实现方法进行了全面比较研究。
引用
@article{arxiv.1903.02428,
title = {Fast Graph Representation Learning with PyTorch Geometric},
author = {Matthias Fey and Jan Eric Lenssen},
journal= {arXiv preprint arXiv:1903.02428},
year = {2019}
}
备注
ICLR 2019 (RLGM Workshop)