pytorch 如何查看數(shù)據(jù)類型和大小
問題描述:
查看tensor數(shù)據(jù)大小時(shí)使用了data.shape(),報(bào)錯(cuò):
TypeError: 'torch.Size' object is not callable 或 TypeError: 'tuple' object is not callable。
解決方法:
查看數(shù)據(jù)類型:data.dtype
查看數(shù)據(jù)大?。篸ata.shape
補(bǔ)充:pytorch tensor比較大小 數(shù)據(jù)類型要注意
如下
a = torch.tensor([[0, 0], [0, 0]]) print(a>=0.5)
輸出
tensor([[1, 1], [1, 1]], dtype=torch.uint8)
結(jié)果明顯不對, 分析原因是因?yàn)? a是long類型, 而0.5是float. 0.5會(huì)被轉(zhuǎn)化為 long, 變?yōu)?. 因此結(jié)果會(huì)出錯(cuò), 做出如下修改就可以得到正確答案
正確用法:
a = torch.tensor([[0, 0], [0, 0]]).float() print(a>=0.5)
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持本站。
版權(quán)聲明:本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有,歡迎引用、轉(zhuǎn)載,請保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站,禁止在非maisonbaluchon.cn所屬的服務(wù)器上建立鏡像,否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來,僅供學(xué)習(xí)參考,不代表本站立場,如有內(nèi)容涉嫌侵權(quán),請聯(lián)系alex-e#qq.com處理。
關(guān)注官方微信