交换模块

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

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

Parameters

• mod – 输入模块

• 映射 – 一个从 nn 模块映射到 nnq 模块的字典

Returns

对应的量化模块 mod