仅作为记录，大佬请跳过。

背景

将类似于torch.tensor([64,3,224,224])的tensor转换成图像显示

做法

设a是上述长度的tensor

from torchvision import transforms
unloader = transforms.ToPILImage()
a=a[0]  # 从torch.tensor([64,3,224,224])中选择一个：torch.tensor([3,224,224])
image = a.cpu().clone()  # clone the tensor
image = unloader(image)
image.save('example.jpg')

参考

传送门