pytorch下只打印tensor的數(shù)值不打印出device等信息怎么辦？

我們在使用pytorch的時候會涉及到打印tensor的值，在pytorch怎么打印tensor的數(shù)值這方面只要會使用pytorch一般都會使用了，但是有些情況下我們還需要打印device的信息，那么pytorch怎么打印device信息呢？接下來的這篇文章告訴你！

torch.Tensor類型的數(shù)據(jù)loss和acc打印時

如果寫成以下寫法

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

如果僅想打印出數(shù)值，使得打印出的信息更加簡潔

則要用以下寫法

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

該寫法還可控制輸出格式，結(jié)果類似如下：

 batch loss: 2.312 batch acc: 0.164

以上就是pytorch怎么打印device信息的全部內(nèi)容，希望能給大家一個參考，也希望大家多多支持W3Cschool。