torch.compiler.is_exporting

torch.compiler.is_exporting()

Indicated whether we’re under exporting.

It’s stricter than is_compiling() flag, as it would only be set to True when torch.export is used.

示例

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

返回类型

布尔值