pytorch plot tensor image in rgb code example Example: pytorch plt.imshow plt.imshow(images[0].permute(1, 2, 0))