PIL转tensor再传PIL并打印图片

import torch.nn as nn
from torchvision import transforms
from PIL import Image
import torch


img = Image.open('../data/pikachu.jpg')

transform_to_tensor = transforms.ToTensor()
img_ = transform_to_tensor(img)

print(1, img_.size())

img_ = torch.unsqueeze(img_, 0)
conv = nn.Conv2d(3, 3, kernel_size=7, padding=3)
img_ = conv(img_)
img_ = conv(img_)
img_ = conv(img_)
img_ = conv(img_)
img_ = torch.squeeze(img_, 0)
print(2, img_.size())

unloader = transforms.ToPILImage()
out = unloader(img_)

out.show()