GraphNeuralNetworks.jl
GraphNeuralNetworks.jl copied to clipboard
Pooling support for `TemporalSnapshotsGNNGraph`
As described for heterographs in the issue #388, TemporalSnapshotGNNGraph
s also need adapted pooling.
List of existing poolings that need to be extended:
- [ ] GlobalPool
- [ ] GlobalAttentionPool
- [ ] TopKPool
- [ ] Set2Set
I will be working on this in the near future.