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。