python - How do I display a single image in PyTorch ...
04.12.2018 · I want to display a single image loaded using ImageLoader and stored in a PyTorch Tensor. When I try to display it via plt.imshow(image) I get: TypeError: Invalid dimensions for image data The .shape of the tensor is: …