View source on GitHub |
This module implements various graph pooling ops.
Functions
pool(...)
: Implements graph pooling.
unpool(...)
: Graph upsampling by inverting the pooling map.
upsample_transposed_convolution(...)
: Graph upsampling by transposed convolution.