torch.compiler.is_compiling

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

表示一个图是否作为 torch.compile()或 torch.export()的一部分执行/追踪。

注意，最终应该弃用另外两个相关标志：

• torch._dynamo.external_utils.is_compiling()

• torch._utils.is_compiling()

示例：

>>> def forward(self, x):
>>>     if not torch.compiler.is_compiling():
>>>        pass # ...logic that is not needed in a compiled/traced graph...
>>>
>>>     # ...rest of the function...

返回类型:

布尔型