How to reshape an Image in pytorch
If you want to convert to grayscale you could do this:
image.mean(dim=1)
I have an image of shape (32, 3, 32, 32). I know it's of the form (batch_size, Channel, Height, Width).
Q. How do I convert it to be (32, 32, 32) overriding the Channel?