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

文档