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.