torch.from_numpy

torch.from_numpy(ndarray) → Tensor

Создаёт Tensor из numpy.ndarray.

Возвращаемый тензор и ndarray разделяют одну и ту же память. Изменения в тензоре будут отражены в ndarray и наоборот. Возвращаемый тензор нельзя изменять размер.

В настоящее время поддерживаются ndarray с типами данных numpy.float64, numpy.float32, numpy.float16, numpy.complex64, numpy.complex128, numpy.int64, numpy.int32, numpy.int16, numpy.int8, numpy.uint8, и numpy.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])