這是我的程式碼

def imshow(inp, title=None):
    inp = inp.numpy().transpose((1, 2, 0))
    mean = np.array([(0.5, 0.5, 0.5)])
    std = np.array([0.953, 0.898, 1.167])
    inp = std * inp + mean
    inp = np.clip(inp, 0, 1)   #不知道作用是什麼
    plt.imshow(inp)
    if title is not None:
        plt.title(title)
    plt.pause(10)
 

經過檢查發現，我在前面將圖片進行16張打包，所以這裡的顯示報錯axes不匹配，於是我加上這句out = torchvision.utils.make_grid(inputs)，再進行讀取檔案就行了。