Shulou(Shulou.com)06/01 Report--

Editor to share with you how to use plt to display tensor in pytorch, I believe most people do not know much about it, so share this article for your reference, I hope you can learn a lot after reading this article, let's go to know it!

problem

The tensor structure of the image is (C), while the format of the picture that can be displayed by plt is (C). C is the number of color channels, and there can be no.

So the problem is to put Tensor (CmaeWJO W) = > numpy (HMWMagol C)

Solution def transimg (img): img = img / 2 + 0.5 # unnormalize npimg = img.numpy () npimg1 = np.transpose (npimg, (1Jing 2Let0)) # C*H*W = > H*W*C return npimg1 is all the content of this article "how to display tensor using plt in pytorch". Thank you for reading! I believe we all have a certain understanding, hope to share the content to help you, if you want to learn more knowledge, welcome to follow the industry information channel!

Welcome to subscribe "Shulou Technology Information " to get latest news, interesting things and hot topics in the IT industry, and controls the hottest and latest Internet news, technology news and IT industry trends.