熔合 FX ¶

熔合模块，如 conv+bn、conv+bn+relu 等，模型必须处于评估模式。融合规则定义在 torch.ao.quantization.fx.fusion_pattern.py

熔合模块，如 conv+bn、conv+bn+relu 等，模型必须处于评估模式。融合规则定义在 torch.ao.quantization.fx.fusion_pattern.py

参数:

• 模型 (*) – 一个 torch.nn.Module 模型

• fuse_custom_config (*) – fuse_fx 的自定义配置。更多详情见 FuseCustomConfig

返回类型:

GraphModule

示例：

from torch.ao.quantization import fuse_fx
m = Model().eval()
m = fuse_fx(m)