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。