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