换模模块 ¶

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

如果模块具有量化的对应模块并且附加了观察器，则交换模块。

参数:

• mod – 输入模块

• 映射 - 从 nn 模块到 nnq 模块的字典

返回值:

相应的量化模块