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。