[32, 160, 256] is not a valid image shape so it won’t work. You could create 32 grayscale images from this tensor but not a single image without reducing the channel dimension. It’s also helpful to post code snippets instead of screenshots.
[32, 160, 256]