TypeError: Bentuk tidak valid (3, 32, 32) untuk data gambar

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