torch.utils.mobile_optimizer#

建立日期：2025 年 6 月 11 日 | 最後更新日期：2025 年 6 月 11 日

PyTorch Mobile 不再積極維護。正在重定向至 ExecuTorch 文件。

警告

PyTorch Mobile 不再積極維護。請檢視 PyTorch 全新裝置端推理庫 ExecuTorch。您還可以查閱關於 XNNPACK 和 Vulkan 委託的文件。

torch.utils.mobile_optimizer.optimize_for_mobile(script_module, optimization_blocklist=None, preserved_methods=None, backend='CPU')[原始碼]#

為移動部署最佳化 torch script 模組。

引數

script_module (ScriptModule) – 型別為 ScriptModule 的 torch script 模組例項。
optimization_blocklist (Optional[set[torch._C._MobileOptimizerType]]) – 型別為 MobileOptimizerType 的集合。當未傳遞集合時，最佳化方法將執行所有最佳化器傳遞；否則，最佳化器方法將執行不包含在 optimization_blocklist 中的最佳化傳遞。
preserved_methods (Optional[list[~AnyStr]]) – 在呼叫 freeze_module 傳遞時需要保留的方法列表
backend (str) – 用於執行結果模型的裝置型別（“CPU”（預設）、“Vulkan”或“Metal”）。

返回

一個新的最佳化的 torch script 模組

返回型別

RecursiveScriptModule

文件