torch.from_numpy

torch.from_numpy(ndarray) → 张量

从 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])