温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

pytorch下怎么只打印tensor的数值不打印device等信息

发布时间:2022-02-24 17:17:14 来源:亿速云 阅读:880 作者:iii 栏目:开发技术

这篇“pytorch下怎么只打印tensor的数值不打印device等信息”文章的知识点大部分人都不太理解,所以小编给大家总结了以下内容,内容详细,步骤清晰,具有一定的借鉴价值,希望大家阅读完这篇文章能有所收获,下面我们一起来看看这篇“pytorch下怎么只打印tensor的数值不打印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下怎么只打印tensor的数值不打印device等信息”这篇文章的内容,相信大家都有了一定的了解,希望小编分享的内容对大家有帮助,若想了解更多相关的知识内容,请关注亿速云行业资讯频道。

向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI