這篇文章將為大家詳細(xì)講解有關(guān)怎么在pytorch中使用 to方法轉(zhuǎn)換類型，文章內(nèi)容質(zhì)量較高，因此小編分享給大家做個(gè)參考，希望大家閱讀完這篇文章后對(duì)相關(guān)知識(shí)有一定的了解。

成都創(chuàng)新互聯(lián)成立與2013年，先為嘉蔭等服務(wù)建站，嘉蔭等地企業(yè)，進(jìn)行企業(yè)商務(wù)咨詢服務(wù)。為嘉蔭企業(yè)網(wǎng)站制作PC+手機(jī)+微官網(wǎng)三網(wǎng)同步一站式服務(wù)解決您的所有建站問題。

將一個(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è)代碼會(huì)出錯(cuò)的。因?yàn)閍和b是不同的device，一個(gè)是CPU，一個(gè)是GPU，不能運(yùn)行。

修改如下：

a = a.to(b)
d = torch.matmul(a, b)
print(type(d))

關(guān)于怎么在pytorch中使用 to方法轉(zhuǎn)換類型就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，可以學(xué)到更多知識(shí)。如果覺得文章不錯(cuò)，可以把它分享出去讓更多的人看到。

網(wǎng)頁標(biāo)題：怎么在pytorch中使用to方法轉(zhuǎn)換類型-創(chuàng)新互聯(lián)
文章地址：http://jinyejixie.com/article20/hgpjo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供做網(wǎng)站、App設(shè)計(jì)、服務(wù)器托管、用戶體驗(yàn)、網(wǎng)站建設(shè)、網(wǎng)站設(shè)計(jì)公司

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請(qǐng)盡快告知，我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如需處理請(qǐng)聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來源： 創(chuàng)新互聯(lián)