FuseCustomConfig

class torch.ao.quantization.fx.custom_config.FuseCustomConfig

用于 fuse_fx() 的自定义配置。

使用示例

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

classmethod from_dict(fuse_custom_config_dict)

从包含以下项的字典创建 ConvertCustomConfig

“preserved_attributes”：即使属性在 forward 中未使用，也会保留的属性列表

此函数主要用于向后兼容，将来可能会删除。

返回类型

FuseCustomConfig

set_preserved_attributes(attributes)

设置即使模型 forward 方法中未使用，也将在图形模块中保留的属性名称。

返回类型

FuseCustomConfig

to_dict()

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

返回类型

Dict[str, Any]