TypeError: Invalid shape (32, 128, 1) for image data site:stackoverflow.com code example

Example: TypeError: Invalid shape (3, 32, 32) for image data

plt.imshow(data[0][0].view(32,32,3))
plt.show()