Hi,
Any suggestion to display an image having a dimension of (128, 3, 32, 32) in Pytorch,
It generates the following error. I think batch size (128) should be removed but how?
Invalid shape (128, 3, 32, 32) for image data