tensor轉成numpy的幾種情況

2021-10-23 15:16:38 字數 350 閱讀 5716

下面將將tensor轉成numpy的幾種情況

gpu中的variable變數:

a.cuda().data.cpu().numpy()

gpu中的tensor變數:

a.cuda().cpu().numpy()

cpu中的variable變數:

a.data.numpy()

cpu中的tensor變數:

a.numpy()

總結:.cuda()是讀取gpu中的資料

.data是讀取variable中的tensor

.cpu是把資料轉移到cpu上

.numpy()把tensor變成numpy

**:[

pytorch的tensor與numpy陣列共享值

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

Pytorch之Tensor和Numpy之間的轉換

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

numpy與tensor中的陣列互轉

如 import torch as t import numpy as np x t.ones 5 float tensor型別為floattensor,也可呼叫long 方法轉為longtensor y x.numpy print x print y 輸出 tensor 1.1.1.1.1.1.1...