TypeError: Invalid shape (3, 112, 112) for image data code example Example: TypeError: Invalid shape (3, 32, 32) for image data plt.imshow(data[0][0].view(32,32,3)) plt.show()