torch 定義了九種cpu tensor型別和九種gpu tensor型別
data type
dtype
cpu tensor
gpu tensor
32位浮點型
torch.float32或torch.float
torch.floattensor
torch.cuda.floattensor
64位浮點型
torch.float64或torch.double
torch.doubletensor
torch.cuda.doubletensor
16位浮點型
torch.float16或torch.half
torch.halftensor
torch.cuda.halftensor
8位整型 (無符號)
torch.uint8
torch.bytetensor
torch.cuda.bytetensor
8位整型 (有符號)
torch.int8
torch.chartensor
torch.cuda.chartensor
16位整型 (有符號)
torch.int16或torch.short
torch.shorttensor
torch.cuda.shorttensor
32位整型 (有符號)
torch.int32或torch.int
torch.inttensor
torch.cuda.inttensor
64位整型 (有符號)
torch.int64或torch.long
torch.longtensor
torch.cuda.longtensor
布林型torch.bool
torch.booltensor
torch.cuda.booltensor
torch.tensor是預設的tensor型別 (torch.floattensor)的別名。
Pytorch資料型別轉換
pytorch的型別可以分為cpu和gpu上的tensor,它們擁有的資料型別是基本上是一樣的 tensor.floattensor tensor.longtensor tensor.bytetensor tensor.chartensor tensor.shorttensor tensor.int...
pytorch 張量 張量的資料型別
張量定義 import torch torch.tensor 1.2 3.4 dtype 獲取張量的資料型別，其中torch.tensor 函式生成乙個張量 torch.float32 torch.set default tensor type torch.doubletensor 設定張量的預設資...
PyTorch的資料型別與建立
資料對比 python pytorch intinttensor of size float floattensor of size int array inttensor of size d1,d2,float floattensor of size d1,d2,string 沒有對string的...