torch.compiler.is_exporting

torch.compiler.is_exporting() [source]

Показывает, выполняется ли экспорт.

Это более строгая проверка, чем флаг is_compiling(): она принимает значение True только при использовании torch.export.

Пример:

>>> def forward(self, x):
>>>     if not torch.compiler.is_exporting():
>>>        pass # ...logic that is not needed in export...
>>>
>>>     # ...rest of the function...

Тип возвращаемого значения:

bool