溫馨提示×

您好，登錄后才能下訂單哦！

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

怎么在pytorch中使用 to方法轉(zhuǎn)換類型

發(fā)布時(shí)間：2021-03-22 16:40:44 來源：億速云 閱讀：144 作者：Leah 欄目：開發(fā)技術(shù)

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

將一個(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))

怎么在pytorch中使用 to方法轉(zhuǎn)換類型

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

向AI問一下細(xì)節(jié)

免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI
助
手