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。