torch.compiler.is_compiling¶
- torch.compiler.is_compiling()[源代码]¶
指示图是否作为 torch.compile() 或 torch.export() 的一部分执行/跟踪。
- 请注意,还有另外 2 个相关的标志,最终应该弃用
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...
- 返回类型