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