@ptrblck thanks for the comment. I have made the changes  to the post. The original size of the image is 3840X1920X3.
Can u plz tell me why for me I can’t pass a tensor of rectangular shape through convolution batchnorm maxpool layers?