pytorch之tensor型別間的相互轉換

2021-09-19 06:25:32 字數 938 閱讀 3750

import torch

tensor = torch.randn(3, 5)

print(tensor)

# torch.long() 將tensor投射為long型別

long_tensor = tensor.long()

print(long_tensor)

# torch.half()將tensor投射為半精度浮點型別

half_tensor = tensor.half()

print(half_tensor)

# torch.int()將該tensor投射為int型別

int_tensor = tensor.int()

print(int_tensor)

# torch.double()將該tensor投射為double型別

double_tensor = tensor.double()

print(double_tensor)

# torch.float()將該tensor投射為float型別

float_tensor = tensor.float()

print(float_tensor)

# torch.char()將該tensor投射為char型別

char_tensor = tensor.char()

print(char_tensor)

# torch.byte()將該tensor投射為byte型別

byte_tensor = tensor.byte()

print(byte_tensor)

# torch.short()將該tensor投射為short型別

short_tensor = tensor.short()

print(short_tensor)

pytorch: tensor型別的構建與相互轉換

Pytorch之Tensor和Numpy之間的轉換

最重要的區別t.tensor和t.tensor 不論輸入的型別是什麼,t.tensor 都會進行資料拷貝,不會共享記憶體 t.tensor 與numpy共享記憶體,但當numpy的資料型別和tensor的型別不一樣的時候,資料會被複製,不會共享記憶體。可使用t.from numpy 或者t.deta...

Pytorch快速入門一 Tensor

tensor 可以簡單地認為是乙個陣列,且支援高效的科學計算。基礎操作 從儲存角度講,可以分為以下兩類 tensor和numpy的相互轉換。tensor 和numpy共享記憶體,所以互相轉換的速度非常快。這也意味著,如果其中乙個變了,另外乙個也會變。import torch as t a t.one...

pytorch的tensor與numpy陣列共享值

網上的很多部落格說tensor與numpy陣列共享記憶體,這是錯誤的 先給結論 tensor與numpy陣列共享部分記憶體,說共享值更嚴謹,且有條件 看 a torch.ones 2,2 b a.numpy print id a print id b 輸出 3030786996336 3030755...