plot torch image code example

Example: pytorch plt.imshow

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