torch.compiler.is_dynamo_compiling¶
- torch.compiler.is_dynamo_compiling()[源代码]¶
指示是否通过 TorchDynamo 追踪图形。
它比 is_compiling() 标志更严格,因为只有在使用 TorchDynamo 时它才会设置为 True。
示例
>>> def forward(self, x): >>> if not torch.compiler.is_dynamo_compiling(): >>> pass # ...logic that is not needed in a TorchDynamo-traced graph... >>> >>> # ...rest of the function...
- 返回类型