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。