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.