torch.utils.mobile_optimizer#
创建日期：2025 年 6 月 11 日 | 最后更新日期：2025 年 6 月 11 日
PyTorch Mobile 不再积极维护。重定向到 ExecuTorch 文档。
警告
PyTorch Mobile 不再积极维护。请查看 ExecuTorch，这是 PyTorch 全新的设备端推理库。您还可以查阅 XNNPACK 和 Vulkan 委托的文档。
- torch.utils.mobile_optimizer.optimize_for_mobile(script_module, optimization_blocklist=None, preserved_methods=None, backend='CPU')[source]#
为移动部署优化 Torch 脚本模块。
- 参数
script_module (ScriptModule) – Torch 脚本模块的实例，类型为 ScriptModule。
optimization_blocklist (Optional[set[torch._C._MobileOptimizerType]]) – 类型为 MobileOptimizerType 的集合。当未传入集合时，优化方法将运行所有优化过程；否则，优化方法将运行优化黑名单中未包含的优化过程。
preserved_methods (Optional[list[~AnyStr]]) – 在调用 freeze_module 过程时需要保留的方法列表
backend (str) – 用于运行结果模型的设备类型（“CPU”（默认）、“Vulkan” 或 “Metal”）。
- 返回
一个经过优化后的新 Torch 脚本模块。
- 返回类型
RecursiveScriptModule