pytorch使用 to 進行類型轉(zhuǎn)換方式
在程序中，有多種方法進行強制類型轉(zhuǎn)換。
本博文將介紹一個非常常用的方法：to（）方法。
我們通常使用它來進行GPU和CPU的類型轉(zhuǎn)換，但其實也可以用來進行torch的dtype轉(zhuǎn)換。
常見方法：tensor.to(‘cuda:0')
先看官網(wǎng)介紹：
**Performs Tensor dtype and/or device conversion. A torch.dtype and torch.device are inferred from the arguments of self.to(*args, kwargs).
本文舉一個例子，將一個tensor轉(zhuǎn)化成與另一個tensor相同的數(shù)據(jù)類型和相同GPU或CPU類型
import torch device = 'cuda:0' a = torch.zeros(2, 3) print(type(a)) b = torch.ones(3, 4).to(device) print(type(b)) c = torch.matmul(a, b) print(type(c))
我們看到這個代碼會出錯的。因為a和b是不同的device，一個是CPU，一個是GPU，不能運行。
修改如下：
a = a.to(b) d = torch.matmul(a, b) print(type(d))
可以看到to還是很好用的，尤其是不確定我們的數(shù)據(jù)類型和device時。
其實pytorch中還有很多其他方法可以這么做，以后會繼續(xù)介紹。
以上這篇pytorch使用 to 進行類型轉(zhuǎn)換方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python json 錯誤xx is not JSON serializable解決辦法
這篇文章主要介紹了Python json 錯誤xx is not JSON serializable解決辦法的相關(guān)資料,需要的朋友可以參考下2017-03-03
Python pandas.DataFrame調(diào)整列順序及修改index名的方法
這篇文章主要介紹了Python pandas.DataFrame調(diào)整列順序及修改index名的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2019-06-06
Python使用Opencv打開筆記本電腦攝像頭報錯解問題及解決
這篇文章主要介紹了Python使用Opencv打開筆記本電腦攝像頭報錯解問題及解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-06-06
這篇文章主要介紹了python解決OpenCV在讀取顯示圖片的時候閃退的問題，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-02-02
淺談keras通過model.fit_generator訓練模型(節(jié)省內(nèi)存)
這篇文章主要介紹了淺談keras通過model.fit_generator訓練模型(節(jié)省內(nèi)存)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
django-simple-captcha多種驗證碼的實現(xiàn)方法
本文介紹了如何在Django項目中配置和使用不同類型的驗證碼,包括數(shù)字驗證碼、字母驗證碼和算術(shù)驗證碼,每種驗證碼結(jié)合實例代碼給大家介紹得非常詳細,感興趣的朋友跟隨小編一起看看吧2024-12-12
Python調(diào)用edge-tts實現(xiàn)在線文字轉(zhuǎn)語音效果
edge-tts是一個 Python 模塊,允許通過Python代碼或命令的方式使用 Microsoft Edge 的在線文本轉(zhuǎn)語音服務(wù),這篇文章主要介紹了Python調(diào)用edge-tts實現(xiàn)在線文字轉(zhuǎn)語音效果,需要的朋友可以參考下2024-03-03
Python用正則表達式實現(xiàn)爬取古詩文網(wǎng)站信息
這篇文章主要給大家介紹了關(guān)于Python如何利用正則表達式爬取爬取古詩文網(wǎng)站信息，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-12-12
今天小編就為大家分享一篇OpenCV+python手勢識別框架和實例講解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-08-08
Window系統(tǒng)下Python如何安裝OpenCV庫
這篇文章主要介紹了Window系統(tǒng)下Python如何安裝OpenCV庫，本文分步驟給大家介紹的詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-03-03