將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維度的位置