我們?cè)谑褂胮ytorch的時(shí)候會(huì)涉及到打印tensor的值,在pytorch怎么打印tensor的數(shù)值這方面只要會(huì)使用pytorch一般都會(huì)使用了,但是有些情況下我們還需要打印device的信息,那么pytorch怎么打印device信息呢?接下來(lái)的這篇文章告訴你!
torch.Tensor類(lèi)型的數(shù)據(jù)loss和acc打印時(shí)
如果寫(xiě)成以下寫(xiě)法
print('batch_loss: '+str(loss.data)+'batch acc: '+str(acc.data))
則不僅會(huì)打印出loss和acc的值,還會(huì)打印出device信息和 tensor字樣,如下:
batch_loss: tensor(2.4252, device='cuda:0')batch acc: tensor(0.2969, device ='cuda:0')
如果僅想打印出數(shù)值,使得打印出的信息更加簡(jiǎn)潔
則要用以下寫(xiě)法
print('batch_loss: {:.3f} batch acc: {:.3f}'.format(loss.data, acc.data))
該寫(xiě)法還可控制輸出格式,結(jié)果類(lèi)似如下:
batch loss: 2.312 batch acc: 0.164
以上就是pytorch怎么打印device信息的全部?jī)?nèi)容,希望能給大家一個(gè)參考,也希望大家多多支持W3Cschool。