So the method of loading the image using opencv and then convert it to tensor is wrong? Is there another way to convert the image to tensor so that it outputs a shape with the dimensions you specified?
also when I only do dim1 and dim2 I get a size of this :
torch.Size([1, 18, 32, 3, 32, 32])


and when I show the image using plt, there’s still something wrong