pytorch常用資料型別所佔位元組數對照表一覽

2022-10-04 12:57:10 字數 1475 閱讀 8791

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...