just use:
train_dataset[0][0].show()
if you insist on using plt.imshow, you 'll have to convert your image to numpy, so that’s another story. Something like:
plt.imshow(np.array(train_dataset[0][0]).squeeze()); plt.show()