batchsize * 1 * 28 * 28

for idx in range(1):
    im_data = images[idx]
    print(im_data.shape)
    im_label = labels[idx]
    im_pred = pred[idx]
    im_data = im_data.transpose(1, 2, 0)
    print(im_data.shape)
    print("label", im_label)
    print("pred",im_pred)
    cv2.imshow("imdate", im_data)
    cv2.waitKey(0)

是不是分别对应维度的位置啊，如果我原来数据的维度是 1 * 28 * 28
然后经过transpose(2, 1, 0)之后就变成28 * 28 * 1
或者是说transpose(1, 0, 2)之后就变成28 * 1 * 28了
那这样的话，如果是经过transpose(2, 1, 0)之后，所得到的28 * 28 * 1那么最后展示出来的图片结果，会不会原来图片的长和宽的位置发生颠倒