Solved by Nikronic in post #4 
              
              
                You misunderstood channels with sizes. In Conv2d, you define input/output channel and kernel size and some arbitrary args like padding, not output size. Output size will be determined using kernel_size, stride, padding, etc. 
https://pytorch.org/docs/master/generated/torch.nn.Conv2d.html 
An image i…