data type
dtype
cpu tensor
gpu tensor
size/bytes
pevkzbw
32-bit floating
torch.float32 or torch.float
torch.floattensor
torch.cuda.floattensor
464-bit floating
torch.float64 or torch.double
torch.doubletensor
torch.cuda.doubletensor
816-bit floating
torch.float16or torch.half
torch.halftensor
torch.cuda.halftensor
-8-bit integer (unsigned)
torch.uint8
torch.bytetensor
torch.cuda.bytetensor
18-bit integer (signed)
torch.int8
torch.chartensor
torch.cuda.chartenwww.cppcns.comsor
-16-bit integer (signed)
torch.int16or torch.short
torch.shorttensor
torch.cuda.shorttensor
232-bit integer (signed)
torch.int32 or torch.int
torch.inttensor
torch.cuda.inttensor
464-bit integer (signed)
torch.int64 or torch.long
torch.longtensor
torch.cuda.longtensor
8以上pytorch中的資料型別和numpy中的相對應,占用位元組大小也是一樣的
補充:pytorch tensor比較大小 資料型別要注意
a = torch.tensor([[0, 0], [0, 0]])
print(a>=0.5)
輸出tensor([[1, 1],
[1, 1]], dtype=torpevkzbwch.uint8)
結果明顯不對, 分析原因是因為, a是long型別, 而0.5是float. 0.5會被轉化為 long, 變為0. 因此結果會出錯, 做出如下修改就可以得到正確答案
a = torch.tensor([[0, 0], [0, 0]]).float()
print(a>=0.5)
本文標題: pytorch常用資料型別所佔位元組數對照表一覽
本文位址: /jiaoben/python/393304.html
pytorch 資料型別
torch 定義了九種cpu tensor型別和九種gpu tensor型別 data type dtype cpu tensor gpu tensor 32位浮點型 torch.float32或torch.float torch.floattensor torch.cuda.floattensor...
Pytorch資料型別轉換
pytorch的型別可以分為cpu和gpu上的tensor,它們擁有的資料型別是基本上是一樣的 tensor.floattensor tensor.longtensor tensor.bytetensor tensor.chartensor tensor.shorttensor tensor.int...
C 資料型別所佔位元組大小
對於指標型別,因為指標指向的是乙個位址,32位作業系統就是4位元組。64位作業系統那就是8位元組了。int這個型別佔作業系統乙個記憶體單元大小。早先16位作業系統乙個記憶體單元是16位,所以是2個位元組 32位系統乙個記憶體單元是是32位,所以是4位元組 64位作業系統,4位元組。整形加unsign...