fuse_fx

class torch.ao.quantization.quantize_fx.fuse_fx(model, fuse_custom_config=None, backend_config=None)

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

Parameters

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

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

Return type

图模块

示例：

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