快捷方式

swap_module

class torch.ao.quantization.swap_module(mod, mapping, custom_module_class_mapping, use_precomputed_fake_quant=False)[source][source]

如果模組有量化對應物且附加了 observer,則交換該模組。

引數
  • mod – 輸入模組

  • mapping – 從 nn 模組對映到 nnq 模組的字典

返回

mod 的相應量化模組

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源