FuseCustomConfig¶

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

用於 fuse_fx() 的自訂配置。

範例用法

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

classmethod from_dict(fuse_custom_config_dict)[source][source]¶

從具有以下項目的字典建立一個 ConvertCustomConfig

“preserved_attributes”: 一個屬性列表，即使它們沒有在 forward 中使用，也會保留這些屬性

此函數主要用於向後相容性，將來可能會被移除。

set_preserved_attributes(attributes)[原始碼][原始碼]¶

設定將在圖模組中保留的屬性名稱，即使它們沒有在模型的 forward 方法中使用。

將此 FuseCustomConfig 轉換為一個字典，其項目描述於 from_dict() 中。

文件