Graph Neural Network with Hierarchical Pooling for PyTorch: "Hierarchical Graph Representation Learning with Differentiable Pooling". - View it on GitHub