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。
