Solved by ptrblck in post #2 
              
              
                The docs describe each input argument (nn.AdaptiveAvgPool2d, nn.Softmax) so you can see that the former is using the argument as the output_size while the latter uses it as the dim argument. 
In case you are unsure what these arguments do, write a small code snippet to check its usage, e.g. via: 
po…