pytorch使用 to 進行型別轉換方式

2022-09-26 12:54:29 字數 851 閱讀 3206

在程式程式設計客棧中,有多種方法進行強制型別轉換。

本博文將介紹乙個非常常用的方法: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...