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