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')[source]#
优化一个 torch script 模块以进行移动端部署。
- 参数:
script_module (ScriptModule) – 类型为 ScriptModule 的 torch script 模块实例。
optimization_blocklist (set[_MobileOptimizerType] | None) – 类型为 MobileOptimizerType 的集合。当未传递集合时，优化方法将运行所有优化器传递；否则，优化方法将运行未包含在 optimization_blocklist 中的优化传递。
preserved_methods (list[AnyStr] | None) – 在调用 freeze_module 传递时需要保留的方法列表
backend (str) – 用于运行结果模型的设备类型（'CPU'（默认），'Vulkan' 或 'Metal'）。
- 返回:
一个新优化的 torch script 模块
- 返回类型:
RecursiveScriptModule
