pytorch下只打印tensor的數值不打印出device等信息的方法

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))

該寫法還可控制輸出格式,結果類似如下:
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章