I have 32 npy files with the shape (32,160,256) and I want to convert them as png images with the same shape

[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.