When you use transforms.ToTensor(), by default it changes the input arrays from HWC to CHW order.
Is it possible to have ToTensor not do this if my input data are already in the ordering I want?
What can I replace this code with to make it keep the ordering I want and still convert to a torch tensor?
transforms.Compose([transforms.ToTensor()])