TypeError: nieprawidłowy kształt (3, 32, 32) dla danych obrazu

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