交换模块

类torch.ao.quantization.swap_module(mod, mapping, custom_module_class_mapping, use_precomputed_fake_quant=False)[源代码]

如果有对应的量化版本且已附加了观察器，则交换该模块。

参数

• mod - 输入模块

• mapping – 将 nn 模块映射到 nnq 模块的字典

返回值

与mod对应的量化模块