這篇文章將為大家詳細(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ù)解決您的所有建站問(wèn)題。

將一個(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í)。如果覺(jué)得文章不錯(cuò)，可以把它分享出去讓更多的人看到。

本文標(biāo)題：怎么在pytorch中使用to方法轉(zhuǎn)換類型-創(chuàng)新互聯(lián)
網(wǎng)站URL：http://weahome.cn/article/hgpjo.html