for images, labels in train_data: for images, labels in train_data: img = images[0] img = img.numpy() img = np.transpose(img, (1, 2, 0)) plt.imshow(img) plt.show()
train_data
原文：https://www.cnblogs.com/sleven/p/14727536.html
