pytorch tensor与numpy转换

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