Python PyTorch from_numpy用法及代码示例

本文简要介绍python语言中 torch.from_numpy 的用法。

用法:

torch.from_numpy(ndarray) → Tensor

从 numpy.ndarray 创建 Tensor 。

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

它目前接受 ndarray 的 dtypes 为 numpy.float64 , numpy.float32 , numpy.float16 , numpy.complex64 , numpy.complex128 , numpy.int64 , numpy.int32 , numpy.int16 , numpy.int8 , numpy.uint8 和 numpy.bool 。

例子：

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