Torch Geometric Pool: the PyTorch library for pooling in Graph Neural Networks
arXiv:2512.12642v2 Announce Type: replace
Abstract: 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, what they return after…