欧美一区二区三区老妇人-欧美做爰猛烈大尺度电-99久久夜色精品国产亚洲a-亚洲福利视频一区二区

怎么在pytorch中使用to方法轉(zhuǎn)換類型-創(chuàng)新互聯(lián)

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

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

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

我們看到這個代碼會出錯的。因?yàn)閍和b是不同的device，一個是CPU，一個是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)容可以對大家有一定的幫助，可以學(xué)到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。

網(wǎng)頁標(biāo)題：怎么在pytorch中使用to方法轉(zhuǎn)換類型-創(chuàng)新互聯(lián)
文章地址：http://chinadenli.net/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)請盡快告知，我們將會在第一時間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)

外貿(mào)網(wǎng)站制作