Solved by Ibrahim_Yusuf in post #3 
              
              
                Thank you @ptrblck. I figured it out. I had a misconception regarding how the image will look after swapping H and W. Later I realized that it is correct, nothing changes about how the model sees the image. It is a thing about how arrays are represented. An image with height H and width W, when repr…