I have an input image with shape (1, 3, 392, 196) but after calling transforms.ToTensor() it converts it to (1, 3, 196, 392), by swapping the width and height. I want to know whether my model is seeing a rotated version of the input image because this is crucial for my object detection task.