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。