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

这篇"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等信息"这篇文章的内容，相信大家都有了一定的了解，希望小编分享的内容对大家有帮助，若想了解更多相关的知识内容，请关注行业资讯频道。