torch.jit.script_if_tracing

torch.jit.script_if_tracing(fn) [source]

Компилирует fn при первом вызове во время трассировки. torch.jit.script имеет ощутимое время запуска при первом вызове из-за ленивой инициализации многих встроенных функций компилятора. Поэтому его не следует использовать в коде библиотеки. Однако, вы можете захотеть, чтобы части вашей библиотеки работали при трассировке, даже если они используют поток управления. В этих случаях вы должны использовать @torch.jit.script_if_tracing для замены torch.jit.script.

Параметры

fn – Функция для компиляции.

Возвращает

Если вызывается во время трассировки, возвращается ScriptFunction, созданный torch.jit.script. В противном случае возвращается исходная функция fn.