image plot torch code example

Example: pytorch plt.imshow

plt.imshow(images[0].permute(1, 2, 0))