分类：不打印device

2021
07-08

解决pytorch下只打印tensor的数值不打印出device等信息的问题

torch.Tensor类型的数据loss和acc打印时如果写成以下写法print('batch_loss:'+str(loss.data)+'batchacc:'+str(acc.data))则不仅会打印出loss和acc的值，还会打印出device信息和tensor字样，如下：如果仅想打印出数值，使得打印出的信息更加简洁则要用以下写法print('batch_loss:{:.3f}batchacc:{:.3f}'.format(loss.data,acc.data))该写法还可控制输出格式，结果类似如下：补充：关于pytorch中用plt显示tensor的问题（te... 继续阅读 >