pytorch展示Tensor类型的图片

from matplotlib import pyplot as plt
# loader使用torchvision中自带的transforms函数
from torchvision import transforms

loader = transforms.Compose([
    transforms.ToTensor()])
unloader = transforms.ToPILImage()


def imshow(tensor, title=None):
    image = tensor.cpu().clone()
    image = image.squeeze(0)
    image = unloader(image)
    plt.imshow(image)
    if title is not None:
        plt.title(title)
        plt.pause(0.001)