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。