我有一个 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')