📜  Python PyTorch from_numpy()

📅  最后修改于: 2022-05-13 01:54:56.837000             🧑  作者: Mango

Python PyTorch from_numpy()

PyTorch 是 Facebook 开发的开源机器学习库。它用于深度神经网络和自然语言处理目的。

函数torch.from_numpy()支持将 numpy 数组转换为 PyTorch 中的张量。它期望输入为 numpy 数组(numpy.ndarray)。输出类型是张量。返回的张量和 ndarray 共享相同的内存。返回的张量不可调整大小。

它目前接受 ndarray,dtype 为 numpy.float64、numpy.float32、numpy.float16、numpy.int64、numpy.int32、numpy.int16、numpy.int8、numpy.uint8 和 numpy.bool。

代码#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 中,反之亦然。