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

Most people do not understand the knowledge points of this article "how to print only the value of tensor under pytorch without printing information such as device", so the editor summarizes the following contents to you, detailed contents, clear steps, and has a certain reference value. I hope you can get something after reading this article. Let's take a look at this "how to print only the value of tensor under pytorch without printing device and other information" article.

When printing data of torch.Tensor type loss and acc

If it is written as follows

Print ('batch_loss:' + str (loss.data) + 'batch acc:' + str (acc.data))

Not only the values of loss and acc will be printed, but also the device information and the words tensor will be printed, as follows:

Batch_loss: tensor (2.4252, device='cuda:0') batch acc: tensor (0.2969, device='cuda:0')

If you only want to print out numerical values, make the printed information more concise.

It should be written in the following ways

Print ('batch_loss: {: .3f} batch acc: {: .3f}' .format (loss.data, acc.data))

This method can also control the output format, and the result is similar to the following:

Batch loss: 2.312 batch acc: 0.164

The above is about the content of this article on "how to print only the value of tensor but not device under pytorch". I believe we all have a certain understanding. I hope the content shared by the editor will be helpful to you. If you want to know more about the relevant knowledge, please pay attention to 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.