快捷方式

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

bool

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源