Torch Geometric Pool – the library for hierarchical GNNs

(github.com)

2 points | by fmb404 10 hours ago ago

1 comments

  • fmb404 10 hours ago

    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.