Python PyTorch from_numpy()用法及代碼示例

PyTorch是由Facebook開發的開源機器學習庫。它用於深度神經網絡和自然語言處理。

函數torch.from_numpy()提供支持將numpy數組轉換為PyTorch中的張量。它期望輸入為numpy數組(numpy.ndarray)。輸出類型為張量。返回的張量和ndarray共享相同的內存。返回的張量不可調整大小。

當前它接受具有numpy.float64，numpy.float32，numpy.float16，numpy.int64，numpy.int32，numpy.int16，numpy.int8，numpy.uint8和numpy.bool的dtypes的ndarray。

用法:torch.sinh(ndarray)

代碼1：

# Importing the PyTorch library 
import torch 
import numpy 
  
# A numpy array of size 6 
a = numpy.array([1.0, -0.5, 3.4, -2.1, 0.0, -6.5]) 
print(a) 
  
# Applying the from_numpy function and 
# storing the resulting tensor in 't' 
t = torch.from_numpy(a) 
print(t)

輸出：

[ 1.  -0.5  3.4 -2.1  0.  -6.5]
tensor([ 1.0000, -0.5000,  3.4000, -2.1000,  0.0000, -6.5000],
       dtype=torch.float64)

對張量的修改將反映在ndarray中，反之亦然。