快捷方式

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 中

参数
  • 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)

文档

访问 PyTorch 的综合开发者文档

查看文档

教程

获得针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获取问题解答

查看资源