torch.jit.script_if_tracing

torch.jit.script_if_tracing(fn)

在跟踪期间首次调用时编译 fn。

torch.jit.script 在第一次调用时由于许多编译器内置函数的延迟初始化，启动时间不可忽视。因此，您不应在库代码中使用它。然而，您可能希望库的某些部分即使在使用了控制流的情况下也能进行跟踪。在这些情况下，您应该使用 @torch.jit.script_if_tracing 来替代 torch.jit.script。

Parameters

fn – 一个待编译的函数。

Returns

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