torch.compiler.is_exporting

torch.compiler.is_exporting()[source][source]

是否处于导出状态。

它比 is_compiling()标志更严格，因为它只有在使用 torch.export 时才会设置为 True。

示例：

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

返回类型:

布尔型