torch.Tensor.numpy

Tensor.numpy(*, force=False) → numpy.ndarray

Возвращает тензор как массив NumPy ndarray.

Если force равно False (по умолчанию), преобразование выполняется только если тензор находится на CPU, не требует grad, не имеет установленного флага сопряжения и имеет тип и структуру данных, поддерживаемые NumPy. Возвращаемый массив ndarray и тензор будут совместно использовать хранилище, поэтому изменения в тензоре будут отражаться в ndarray и наоборот.

Если force равно True, это эквивалентно вызову t.detach().cpu().resolve_conj().resolve_neg().numpy(). Если тензор не находится на CPU или установлен флаг сопряжения или отрицательного значения, тензор не будет совместно использовать своё хранилище с возвращаемым ndarray. Установка force в значение True может быть полезным сокращением.

Параметры:

force (bool) – если True, ndarray может быть копией тензора вместо всегда совместного использования памяти, по умолчанию False.