快捷方式

torch.jit.script_if_tracing

torch.jit.script_if_tracing(fn)[source][source]

在跟踪 (tracing) 期间首次调用 fn 时编译它。

torch.jit.script 由于许多编译器内建 (compiler builtins) 的延迟初始化,在首次调用时具有不可忽略的启动时间。因此,不应在库代码中使用它。但是,即使您的库代码使用了控制流 (control flow),您可能仍希望其部分功能在跟踪 (tracing) 模式下工作。在这种情况下,应使用 @torch.jit.script_if_tracing 来替代 torch.jit.script

参数

fn – 要编译的函数。

返回

如果在跟踪 (tracing) 期间调用,将返回由 torch.jit.script 创建的 ScriptFunction。否则,将返回原始函数 fn

文档

查阅 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源