Torch Geometric Pool: the PyTorch library for pooling in Graph Neural Networks

Carlo Abate, Ivan Marisca, Filippo Maria Bianchi

Torch Geometric Pool (tgp) is a pooling library built on top of PyTorch Geometric. Graph pooling methods differ in how they assign nodes to supernodes, how they handle batches, wha…