torch.jit.script_if_tracing¶
- torch.jit.script_if_tracing(fn)[source][source]¶
在跟蹤 (tracing) 期間首次呼叫
fn時編譯它。torch.jit.script由於許多編譯器內建 (compiler builtins) 的延遲初始化，在首次呼叫時具有不可忽略的啟動時間。因此，不應在庫程式碼中使用它。但是，即使您的庫程式碼使用了控制流 (control flow)，您可能仍希望其部分功能在跟蹤 (tracing) 模式下工作。在這種情況下，應使用@torch.jit.script_if_tracing來替代torch.jit.script。- 引數
fn – 要編譯的函式。
- 返回
如果在跟蹤 (tracing) 期間呼叫，將返回由 torch.jit.script 建立的
ScriptFunction。否則，將返回原始函式 fn。
