融合自定义配置 ¶

类 torch.ao.quantization.fx.custom_config.FuseCustomConfig[source][source] ¶

fuse_fx() 的定制配置。

演示用法：

fuse_custom_config = FuseCustomConfig().set_preserved_attributes(["attr1", "attr2"])

类方法 from_dict(fuse_custom_config_dict)[source][source] ¶

从以下项目创建一个 ConvertCustomConfig ：

“保留属性”：即使在 forward 中未使用，这些属性也会持续存在

此功能主要用于向后兼容，未来可能会被移除。

返回类型:

FuseCustomConfig

set_preserved_attributes(attributes)

设置在图模块中即使未在模型的 forward 方法中使用也将持久化的属性名称。

返回类型:

融合自定义配置

to_dict()[来源][来源] ¶

将此 FuseCustomConfig 转换为包含 from_dict() 描述项的字典。

返回类型:

dict[str, Any]