How to generalize Conv2d to MaxPool2d to ConvTranspose2d?

Question

I have generalized a PyTorch autoencoder implementation. One major strategy I\'m using is a rescaling of a sequence which indicates nodes per layer. This way I can run expe