Python 将Pytorch 4D张量另存为图像

我有一个4-d的Pytorch张量，我想把它作为.jpg保存到磁盘上

我的张量大小如下：

print(image_tensor.size())

>>>torch.Size([1, 3, 400, 711])

我可以在IDE中将整个张量视为一个图像：

ax1.imshow(im_convert(image_tensor))

因为我可以把整个张量看作一个图像，所以我假设有一种方法可以把它保存下来。但是，当我尝试保存图像时，它看起来只保存蓝色通道。我想把整个张量保存为一个图像

img1 = image_tensor[0]
save_image(img1, 'img1.jpg')

---

在PyTorch中，此代码段正在工作并保存图像：

from torchvision.utils import save_image
import torch
import torchvision

tensor= torch.rand(2, 3, 400, 711) 

img1 = tensor[0]
save_image(img1, 'img1.png')

---

在保存图像之前，您可以在任何情况下检查

img1的形状。
图像通常具有最后一个颜色通道。