您好,登錄后才能下訂單哦!
本篇文章為大家展示了怎么在pytorch中查看數據類型和大小,內容簡明扼要并且容易理解,絕對能使你眼前一亮,通過這篇文章的詳細介紹希望你能有所收獲。
查看tensor數據大小時使用了data.shape(),報錯:
TypeError: 'torch.Size' object is not callable 或 TypeError: 'tuple' object is not callable。
查看數據類型:data.dtype
查看數據大小:data.shape
補充:pytorch tensor比較大小 數據類型要注意
a = torch.tensor([[0, 0], [0, 0]]) print(a>=0.5)
輸出
tensor([[1, 1], [1, 1]], dtype=torch.uint8)
結果明顯不對, 分析原因是因為, a是long類型, 而0.5是float. 0.5會被轉化為 long, 變為0. 因此結果會出錯, 做出如下修改就可以得到正確答案
a = torch.tensor([[0, 0], [0, 0]]).float() print(a>=0.5)
1.PyTorch是相當簡潔且高效快速的框架;2.設計追求最少的封裝;3.設計符合人類思維,它讓用戶盡可能地專注于實現自己的想法;4.與google的Tensorflow類似,FAIR的支持足以確保PyTorch獲得持續的開發更新;5.PyTorch作者親自維護的論壇 供用戶交流和求教問題6.入門簡單
上述內容就是怎么在pytorch中查看數據類型和大小,你們學到知識或技能了嗎?如果還想學到更多技能或者豐富自己的知識儲備,歡迎關注億速云行業資訊頻道。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。