pytorch 如何查看數(shù)據(jù)類型和大小
問題描述:
查看tensor數(shù)據(jù)大小時使用了data.shape(),報錯:
TypeError: 'torch.Size' object is not callable 或 TypeError: 'tuple' object is not callable。
解決方法:
查看數(shù)據(jù)類型:data.dtype
查看數(shù)據(jù)大?。篸ata.shape
補充:pytorch tensor比較大小 數(shù)據(jù)類型要注意
如下
a = torch.tensor([[0, 0], [0, 0]]) print(a>=0.5)
輸出
tensor([[1, 1],
[1, 1]], dtype=torch.uint8)
結(jié)果明顯不對, 分析原因是因為, a是long類型, 而0.5是float. 0.5會被轉(zhuǎn)化為 long, 變?yōu)?. 因此結(jié)果會出錯, 做出如下修改就可以得到正確答案
正確用法:
a = torch.tensor([[0, 0], [0, 0]]).float() print(a>=0.5)
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
數(shù)據(jù)清洗之如何用一行Python代碼去掉文本中的各種符號
我們在處理文本的時候往往需要對標(biāo)點符號進行處理,下面這篇文章主要給大家介紹了關(guān)于數(shù)據(jù)清洗之如何用一行Python代碼去掉文本中的各種符號的相關(guān)資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-11-11
python 網(wǎng)絡(luò)爬蟲初級實現(xiàn)代碼
這篇文章主要介紹了python 網(wǎng)絡(luò)爬蟲初級實現(xiàn)代碼,需要的朋友可以參考下2016-02-02
Python列表元組字典集合存儲結(jié)構(gòu)詳解
本文詳細介紹了Python中列表、元組、字典和集合等數(shù)據(jù)結(jié)構(gòu)的定義、操作和用法,包括數(shù)據(jù)類型的相互嵌套、常用操作方法、循環(huán)遍歷等2025-02-02

