Tensor <——> Numpy

Tensor與Numpy之間可以相互轉換：
代碼：

[plain]?view plaincopy

import?torch??

import?numpy?as?np??

??

data?=?np.array([[1,2,3,4,5],[6,7,8,9,10]])??

print(data)??

#?numpy?->?tensor??

data_tensor?=?torch.from_numpy(data)??

print(data_tensor)??

#?tensor?->?numpy??

data_numpy?=?data_tensor.numpy()??

print(data_numpy)??

輸出：

[plain]?view plaincopy

[[?1??2??3??4??5]??

[?6??7??8??9?10]]??

tensor([[?1,??2,??3,??4,??5],??

[?6,??7,??8,??9,?10]],?dtype=torch.int32)??

[[?1??2??3??4??5]??

[?6??7??8??9?10]]??

Variable -> Numpy

Variable可以直接轉為Numpy：

[plain]?view plaincopy

import?torch??

from?torch.autograd?import?Variable??

??

#?定義一個Variable??

data?=?Variable(torch.Tensor([4]),requires_grad?=?True)??

#?Variable?->?Numpy??

data_numpy?=?data.detach().numpy()??

print('Numpy:',data_numpy)??

如果把requires_grad改為False，得到的是Tensor：

[plain]?view plaincopy

data_false?=?Variable(data_tensor,requires_grad?=?False)??

print(data_false)??

輸出：

tensor([4.])

Numpy ->Tensor -> Variable

但是Numpy不可以直接轉為Variable，所以要借助Tensor進行轉換：

[plain]?view plaincopy

#?Numpy?-->?Variable?(?Numpy?->Tensor?->?Variable?)??

#?1.?Numpy?->Tensor??

data_tensor?=?torch.from_numpy(data_numpy)??

print('Tensor:',data_tensor)??

#?2.?Tensor?->?Variable??

data_Variable?=?Variable(data_tensor,requires_grad?=?True)??

print('Variable:',data_Variable)??

輸出：

[plain]?view plaincopy

Numpy:?[4.]??

Tensor:?tensor([4.])Variable:?tensor([4.],?requires_grad=True)??

若試圖直接將Numpy轉為Variable，

data_Variable = Variable(data_numpy)，會出現(xiàn)如下錯誤：

來源:https://www.geek-share.com/detail/2774546287.html

總結

以上是生活随笔為你收集整理的Variable、Tensor、Numpy的转换的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。