pytorch下只打印tensor的数值不打印出device等信息怎么办？
- 后端
- 猿友
- 0
我们在使用pytorch的时候会涉及到打印tensor的值，在pytorch怎么打印tensor的数值这方面只要会使用pytorch一般都会使用了，但是有些情况下我们还需要打印device的信息，那么pytorch怎么打印device信息呢？接下来的这篇文章告诉你！
torch.Tensor类型的数据loss和acc打印时
如果写成以下写法
print('batch_loss: '+str(loss.data)+'batch acc: '+str(acc.data))
则不仅会打印出loss和acc的值，还会打印出device信息和 tensor字样，如下：
batch_loss: tensor(2.4252, device='cuda:0')batch acc: tensor(0.2969, device ='cuda:0')
如果仅想打印出数值，使得打印出的信息更加简洁
则要用以下写法
print('batch_loss: {:.3f} batch acc: {:.3f}'.format(loss.data, acc.data))
该写法还可控制输出格式，结果类似如下：
batch loss: 2.312 batch acc: 0.164
以上就是pytorch怎么打印device信息的全部内容，希望能给大家一个参考，也希望大家多多支持W3Cschool。
免责申明：本站发布的内容（图片、视频和文字）以转载和分享为主，文章观点不代表本站立场，如涉及侵权请联系站长邮箱：xbc-online@qq.com进行反馈，一经查实，将立刻删除涉嫌侵权内容。
