torch.jit.script_if_tracing

torch.jit.script_if_tracing(fn)

在跟踪期间第一次调用 fn 时编译它。

torch.jit.script 在第一次调用时会有明显的启动时间，因为许多编译器内置函数的延迟初始化。因此，您不应在库代码中使用它。但是，您可能希望让库的某些部分在跟踪中工作，即使它们使用控制流。在这种情况下，您应该使用 @torch.jit.script_if_tracing 来代替 torch.jit.script。

参数

fn – 要编译的函数。

返回值

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