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