TypeError: неверная форма (3, 32, 32) для данных изображения

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