在程式程式設計客棧中,有多種方法進行強制型別轉換。
本博文將介紹乙個非常常用的方法:to()方法。
我們通常使用它來進行gpu和cpu的型別轉換,但其實也可以用來進行torch的dtype轉換。
常見方法:tensor.to(程式設計客棧『cuda:0')
先看官網介紹:
**performs tensor dtype and/or device conversion. a torch.dtype and t are inferred from the arguments of self.to(*args, kwargs).
本文舉乙個例子,將乙個tensor轉化成與另乙個tensor相同的資料型別和相同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))
我們看到這個**會出錯的。因為a和b是不同的device,乙個是cpu,乙個是gpu,不能執行。
修程式設計客棧改如下:
a = a.to(b)
d = torchxxueub.matmul(a, b)
print(type(d))
可以看到to還是很好用的,尤其是不確定我們的資料型別和device時。
其實pytorch中還有很多其他方法可以這麼做,以後會繼續介紹。
本文標題: pytorch使用 to 進行型別轉換方式
本文位址:
pytorch 使用visdom進行視覺化
相比tensorbordx,visdom重新整理更快,介面體驗也良好,首先是visdom的安裝,與普通的python庫一樣,直接pip install visdom即可 成功安裝後,在控制台下輸入python m visdom.server 複製http localhost 8097,輸入瀏覽器即可...
pytorch 資料型別
torch 定義了九種cpu tensor型別和九種gpu tensor型別 data type dtype cpu tensor gpu tensor 32位浮點型 torch.float32或torch.float torch.floattensor torch.cuda.floattensor...
23 2 2 使用bool型別進行判斷
23.2.2 使用bool型別進行判斷 使用非布林型的數值可以直接參與邏輯判斷和運算 code int ip new int if ip 這樣的 很簡潔漂亮,但是既然c 特意提出了bool型別,那麼建議盡量使用bool型別而非其它值型別來進行邏輯判斷 code int ip new int if i...