读取图片并且显示，运行，numpy报错：axes don't match array

这是我的代码

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)，再进行读取文件就行了。