關(guān)于torch中tensor數(shù)據(jù)類型的轉(zhuǎn)換
更新時間：2022年11月11日 09:42:33 作者：深淵為陵
這篇文章主要介紹了關(guān)于torch中tensor數(shù)據(jù)類型的轉(zhuǎn)換方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
torch中tensor數(shù)據(jù)類型轉(zhuǎn)換
轉(zhuǎn)換為float32類型：
points = points.type(torch.float32)
讀取torch.Tensor數(shù)據(jù)類型
最近弄一下yolov5的detect.py，想提取一下檢測數(shù)據(jù)，記錄
a = torch.tensor(4., device='cuda:0') print(a.item(), a.dtype, a.data, a.int()) ''' 輸出： 4.0 torch.float32 tensor(4., device='cuda:0') tensor(4, device='cuda:0', dtype=torch.int32) ********************************************* a.item() 輸出里面的數(shù)字（yolo中即代表class的下標(biāo)） a.dtype 輸出數(shù)據(jù)類型 a.data 輸出當(dāng)前的數(shù)據(jù)，同print(a) a.int() 數(shù)據(jù)類型轉(zhuǎn)換成整型 '''
torch.Tensor 類型，具體 data 為 tensor(1., device=‘cuda:0’)
跳轉(zhuǎn)看一下類型的定義（ctrl+點擊tensor）
a = torch.tensor()
可以看到定義是
def tensor(data: Any, dtype: Optional[_dtype]=None, device: Union[_device, str, None]=None, requires_grad: _bool=False) -> Tensor: ...
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python根據(jù)給定文件返回文件名和擴(kuò)展名的方法
這篇文章主要介紹了python根據(jù)給定文件返回文件名和擴(kuò)展名的方法,實例分析了Python操作文件及字符串的技巧,非常具有實用價值,需要的朋友可以參考下2015-03-03
Python中class內(nèi)置方法__init__與__new__作用與區(qū)別解析
這篇文章主要介紹了Python中class內(nèi)置方法__init__與__new__作用與區(qū)別探究,本文中涉及的類均為Python3中默認(rèn)的新式類，對應(yīng)Python2中則為顯式繼承了object的class，因為未繼承object基類的舊式類并沒有這些內(nèi)置方法，需要的朋友可以參考下2022-09-09
Django 允許局域網(wǎng)中的機器訪問你的主機操作
這篇文章主要介紹了Django 允許局域網(wǎng)中的機器訪問你的主機實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
這篇文章主要介紹了Python 如何批量更新已安裝的庫，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-05-05
內(nèi)存管理是指在程序的運行過程中，分配內(nèi)容和回收內(nèi)存的過程。如果只分配，不回收，電腦上那點內(nèi)存很快就被用光。幸運的是，Python和Java等高級語言會自動管理內(nèi)存的分配和回收2022-07-07
這篇文章主要介紹了python基于遞歸解決背包問題，遞歸是個好東西，任何具有遞歸性質(zhì)的問題通過函數(shù)遞歸調(diào)用會變得很簡單。一個很復(fù)雜的問題，幾行代碼就能搞定,需要的朋友可以參考下2019-07-07
Pytorch實現(xiàn)LSTM案例總結(jié)學(xué)習(xí)
這篇文章主要介紹了Pytorch實現(xiàn)LSTM案例總結(jié)學(xué)習(xí)，文章通過構(gòu)建網(wǎng)絡(luò)層、前向傳播forward()展開主題介紹，需要的小伙吧可以參考一下2022-07-07
在mac下查找python包存放路徑site-packages的實現(xiàn)方法
今天小編就為大家分享一篇在mac下查找python包存放路徑site-packages的實現(xiàn)方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
好用的Python編輯器WingIDE的使用經(jīng)驗總結(jié)
WingIDE是個專為python程序語言設(shè)計的集成開發(fā)環(huán)境。從1999年起，Wingware公司便開始專注于python開發(fā)，目前WingIDE已經(jīng)是著名的python開發(fā)框架，面向項目風(fēng)格的 IDE 對于大型產(chǎn)品非常有用， 是個很有前途的開發(fā)環(huán)境。2016-08-08
這篇文章主要介紹了Python性能分析工具Profile使用實例,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-11-11