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。