您好,登錄后才能下訂單哦!
這篇“pytorch下怎么只打印tensor的數值不打印device等信息”文章的知識點大部分人都不太理解,所以小編給大家總結了以下內容,內容詳細,步驟清晰,具有一定的借鑒價值,希望大家閱讀完這篇文章能有所收獲,下面我們一起來看看這篇“pytorch下怎么只打印tensor的數值不打印device等信息”文章吧。
如果寫成以下寫法
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等信息”這篇文章的內容,相信大家都有了一定的了解,希望小編分享的內容對大家有幫助,若想了解更多相關的知識內容,請關注億速云行業資訊頻道。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。