torch.from_numpy

torch.from_numpy(ndarray) → Tensor

从一个 numpy.ndarray 创建一个 Tensor 。

返回的张量和 ndarray 共享相同的内存。对张量的修改将在 ndarray 中反映，反之亦然。返回的张量不可调整大小。

目前支持 ndarray ，其数据类型为 numpy.float64 、 numpy.float32 、 numpy.float16 、 numpy.complex64 、 numpy.complex128 、 numpy.int64 、 numpy.int32 、 numpy.int16 、 numpy.int8 、 numpy.uint8 和 bool 。

警告

将写入从只读 NumPy 数组创建的张量不支持，并将导致未定义行为。

示例：

>>> a = numpy.array([1, 2, 3])
>>> t = torch.from_numpy(a)
>>> t
tensor([ 1,  2,  3])
>>> t[0] = -1
>>> a
array([-1,  2,  3])