看程式碼，tensor轉numpy:

a = torch.ones(2,2)
b = a.numpy()
c=np.array(a) #也可以轉numpy陣列
print(type(a))
print(type(b))
print(a)
print(b)

輸出為：

<class ‘torch.Tensor'>
<class ‘numpy.ndarray'>
tensor([[1.,1.],[1.,1.]])
[[1. 1.]
[1. 1.]]

numpy轉tensor：

import torch
import numpy as np

a = np.ones(5)
b = torch.from_numpy(a)
c=torch.Tensor(a) #也可以轉pytorch Tensor
print(type(a))
print(type(b))
print(a)
print(b)

輸出為：

<class ‘numpy.ndarray'>
<class ‘torch.Tensor'>
[1. 1. 1. 1. 1.]
tensor([1.,1.,dtype=torch.float64)

可見pytorch的tensor物件與numpy陣列是可以相互轉換的，且numpy陣列的預設型別是double

以上這篇pytorch 實現tensor與numpy陣列轉換就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。