torch.utils.mobile_optimizer

Создано: 11 июн. 2025 г. | Последнее обновление: 24 июл. 2026 г.

Поддержка PyTorch Mobile больше не осуществляется. Перенаправление на документацию ExecuTorch.

torch.utils.mobile_optimizer.optimize_for_mobile(script_module, optimization_blocklist=None, preserved_methods=None, backend='CPU') [исходный код]

Оптимизирует модуль torch script для развертывания на мобильных устройствах.

Параметры:

• script_module (ScriptModule) – Экземпляр модуля torch script типа ScriptModule.
• optimization_blocklist (set[_MobileOptimizerType] | None) – Множество типа MobileOptimizerType. Если множество не передано, метод оптимизации выполнит все проходы оптимизатора; в противном случае метод оптимизации выполнит проходы оптимизатора, не включённые в optimization_blocklist.
• preserved_methods (list[AnyStr] | None) – Список методов, которые необходимо сохранить при вызове прохода freeze_module
• backend (str) – Тип устройства, на котором будет выполняться результирующая модель («CPU» (по умолчанию), «Vulkan» или «Metal»).

Возвращает:

Новый оптимизированный модуль torch script

Тип возвращаемого значения:

RecursiveScriptModule

torch.utils.mobile_optimizer.generate_mobile_module_lints(script_module) [исходный код]

Создаёт список lint-проверок для указанного модуля torch script.

Параметры:

script_module (ScriptModule) – Экземпляр модуля torch script типа ScriptModule.

Возвращает:

Список словарей, содержащих lint-проверки модулей

Тип возвращаемого значения:

lint_map