快捷方式

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...
返回类型

bool

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源