Solved by AbdulsalamBande in post #10 
              
              
                The input of a Pytorch Neural Network is of type  [BATCH_SIZE] * [CHANNEL_NUMBER] * [HEIGHT] * [WIDTH]. 
Example  : So lets assume you image is of dimension 1×3×32×32 meaning that you have 1 image with 3 channels (RGB) with height 32 and width 32. So using the formular of convolution which is  ((W -…