TypeError: forma inválida (3, 32, 32) para datos de imagen

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