将tensor转换为图像

import torch
from torchvision import transforms
 
toPIL = transforms.ToPILImage() #这个函数可以将张量转为PIL图片，由小数转为0-255之间的像素值
img = torch.randn(3,128,64)
pic = toPIL(img)
pic.save('random.jpg')

注意transforms.ToPILImage()的参数格式：

将形状为 C x H x W 的张量或形状为 H x W x C 的 numpy ndarray 转换为 PIL 图像，同时保留值范围。

可以用permute函数来换tensor维度的位置