fuse_fx

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

合併模組，例如 conv+bn、conv+bn+relu 等，模型必須處於 eval 模式。合併規則定義在 torch.ao.quantization.fx.fusion_pattern.py 中。

參數

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

• fuse_custom_config (*) – fuse_fx 的自定義配置。 詳情請參閱 FuseCustomConfig

回傳類型

GraphModule

範例

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