I have a PyTorch model (PoolNet) that uses an adaptive average pooling layer in the following manner:
ppms = [] for ii in [1, 3, 5]: ppms.append(nn.Sequen