torch.Tensor.numpy

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

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

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

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

Параметры

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