詳解pytorch tensor和ndarray轉(zhuǎn)換相關(guān)總結(jié)
在使用pytorch的時(shí)候，經(jīng)常會(huì)涉及到兩種數(shù)據(jù)格式tensor和ndarray之間的轉(zhuǎn)換，這里總結(jié)一下兩種格式的轉(zhuǎn)換：
1. tensor cpu 和tensor gpu之間的轉(zhuǎn)化：
tensor cpu 轉(zhuǎn)為tensor gpu：
tensor_gpu = tensor_cpu.cuda()
>>> tensor_cpu = torch.ones((2,2))
tensor([[1., 1.],
[1., 1.]])
>>> tensor_gpu = tensor_cpu.cuda()
tensor([[1., 1.],
[1., 1.]], device='cuda:0')
tensor gpu 轉(zhuǎn)為tensor cpu：
tensor_cpu = tensor_gpu.cuda()
>>> tensor_gpu.cpu()
tensor([[1., 1.],
[1., 1.]])
2. tensor cpu 和 ndarray 之間的轉(zhuǎn)化：
tensor cpu 轉(zhuǎn)為 ndarray：
>>> np_array= tensor_cpu.numpy()
array([[1., 1.],
[1., 1.]], dtype=float32)
ndarray 轉(zhuǎn)為 tensor cpu：
注：ndarray的默認(rèn)精度為64位，Tensor的默認(rèn)精度位32位，所以通過(guò)Tensor直接轉(zhuǎn)換的話(huà)，精度會(huì)轉(zhuǎn)換到32位，若通過(guò)from_numpy的方式，則會(huì)保留原來(lái)64位精度
>>> torch.from_numpy(np.ones((2,2)))
tensor([[1., 1.],
[1., 1.]], dtype=torch.float64)
>>> torch.Tensor(np.ones((2,2)))
tensor([[1., 1.],
[1., 1.]])
3. tensor cpu 和 scalar 之間的轉(zhuǎn)化：
如果只是訓(xùn)練了一個(gè)簡(jiǎn)單的分類(lèi)網(wǎng)絡(luò)，對(duì)單個(gè)樣本的輸出會(huì)是一個(gè)標(biāo)量（scalar）
>>>torch.ones((1,1)).item() 1.0
通過(guò)一張圖說(shuō)明三者的轉(zhuǎn)化方式：
到此這篇關(guān)于詳解pytorch tensor和ndarray轉(zhuǎn)換相關(guān)總結(jié)的文章就介紹到這了,更多相關(guān)pytorch tensor和ndarray轉(zhuǎn)換內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
python實(shí)現(xiàn)批量獲取指定文件夾下的所有文件的廠(chǎng)商信息
這篇文章主要介紹了python實(shí)現(xiàn)批量獲取指定文件夾下的所有文件的廠(chǎng)商信息的方法,是非常實(shí)用的技巧,涉及到文件的讀寫(xiě)與字典的操作等技巧,需要的朋友可以參考下2014-09-09
Anaconda出現(xiàn)CondaHTTPError: HTTP 000 CONNECTION FAILED for url
使用anaconda創(chuàng)建一個(gè)新的環(huán)境，執(zhí)行“conda create -n scrapyEnv python=3.6”，結(jié)果出現(xiàn)了CondaHTTPError，下面我們就一起來(lái)了解一下解決方法吧2021-05-05
這篇文章主要為大家介紹了python類(lèi)方法和靜態(tài)方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來(lái)幫助2022-01-01
Redis持久化機(jī)制實(shí)現(xiàn)原理和流程
這篇文章主要介紹了Redis持久化機(jī)制實(shí)現(xiàn)原理和流程,持久化就是把內(nèi)存中的數(shù)據(jù)存放到磁盤(pán)中，防止宕機(jī)后內(nèi)存數(shù)據(jù)丟失,需要的朋友可以參考下2023-04-04
今天小編就為大家分享一篇Pytorch實(shí)現(xiàn)各種2d卷積示例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-12-12
這篇文章主要介紹了Python使用openpyxl復(fù)制整張sheet，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
python函數(shù)裝飾器構(gòu)造和參數(shù)傳遞
這篇文章主要介紹了python函數(shù)裝飾器構(gòu)造和參數(shù)傳遞，下面通過(guò)一個(gè)小案例來(lái)簡(jiǎn)單的理解什么是裝飾器，需要的小伙伴可以參考一下2022-03-03
Python中分支語(yǔ)句與循環(huán)語(yǔ)句實(shí)例詳解
這篇文章主要給大家介紹了關(guān)于Python中分支語(yǔ)句與循環(huán)語(yǔ)句的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家學(xué)習(xí)或者使用python具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2018-09-09
在本篇文章里小編給大家整理的是一篇關(guān)于python3中確保枚舉值代碼分析內(nèi)容，有興趣的朋友們可以學(xué)習(xí)下。2020-12-12
用sqlalchemy構(gòu)建Django連接池的實(shí)例
今天小編就為大家分享一篇用sqlalchemy構(gòu)建Django連接池的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-08-08