tgp is a library that provides a broad suite of graph pooling layers to be inserted into Graph Neural Network architectures built with PyTorch Geometric. With tgp, you can effortlessly construct hierarchical GNNs by interleaving message-passing layers with any pooling operations.
1 comments