torch.from_numpy

torch.from_numpy(ndarray) → Tensor

從 numpy.ndarray 建立一個 Tensor。

回傳的 tensor 和 ndarray 共享相同的記憶體。對 tensor 的修改會反映在 ndarray 中，反之亦然。回傳的 tensor 無法調整大小。

目前接受具有 ndarray 資料類型為 numpy.float64、numpy.float32、numpy.float16、numpy.complex64、numpy.complex128、numpy.int64、numpy.int32、numpy.int16、numpy.int8、numpy.uint8 和 bool 的 ndarray。

警告

不支援寫入從唯讀 NumPy 陣列建立的 tensor，並且會導致未定義的行為。

範例

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