pytorch tensor类型转换_pytorch转caffe

torch.Tensor类型的数据loss和acc打印时 ，如果写成以下写法

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

则不仅会打印出loss和acc的值，还会打印出device信息和 tensor字样，如下：

在这里插入图片描述

如果仅想打印出数值，使得打印出的信息更加简洁，则要用以下写法

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

该写法还可控制输出格式，结果类似如下：

在这里插入图片描述

更多参考

https://www.cnblogs.com/wzyuan/p/9733433.html https://www.cnblogs.com/IaCorse/p/11762548.html https://blog.csdn.net/qq_35260390/article/details/114820089

版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/183395.html原文链接：https://javaforall.cn