pytorch使用 to 進行類型轉(zhuǎn)換方式
更新時間：2020年01月08日 10:24:34 作者：月下花弄影
今天小編就為大家分享一篇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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
python可視化 matplotlib畫圖使用colorbar工具自定義顏色
這篇文章主要介紹了python可視化 matplotlib畫圖使用colorbar工具自定義顏色，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-12-12
Python數(shù)據(jù)結構與算法中的隊列詳解(2)
這篇文章主要為大家詳細介紹了Python中的隊列，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-03-03
Python 由字符串函數(shù)名得到對應的函數(shù)(實例講解)
下面小編就為大家?guī)硪黄狿ython 由字符串函數(shù)名得到對應的函數(shù)(實例講解)。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2017-08-08
大家好，本篇文章主要講的是用python編寫一個圖片拼接工具，感興趣的同學趕快來看一看吧，對你有幫助的話記得收藏一下2022-01-01
這篇文章主要介紹了Python詞云展示庫wordcloud配置詳解,詞云就是將詞語通過圖形可視化的方式，直觀和藝術的展示出來,詞云在我們的生活中經(jīng)常能夠看到，無論是中文的詞云還是英文的詞云,需要的朋友可以參考下2023-07-07
Python 的 httpx 包是一個用于 HTTP 交互的一個優(yōu)秀且靈活的模塊。本文進行詳細的講解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-03-03
在Python中，判斷一個列表（List）是否包含某個特定元素是常見的任務之一，本文就來介紹一下多種判斷List成員包含性的方法，并提供豐富的示例代碼，以幫助大家更好地理解和運用這些技術2023-12-12
這篇文章主要為大家介紹了Python多線程編程，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-12-12
這篇文章主要介紹了pytorch + visdom 處理簡單分類問題的示例，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-06-06
python使用tkinter實現(xiàn)屏幕中間倒計時
這篇文章主要為大家詳細介紹了python使用tkinter實現(xiàn)屏幕中間倒計時，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-03-03