CastConfig¶
- class torchao.float8.CastConfig(scaling_type: ScalingType = ScalingType.DYNAMIC, scaling_granularity: ScalingGranularity = ScalingGranularity.TENSORWISE, target_dtype: Optional[dtype] = None)[source]¶
用於將單個張量轉換為 float8 的配置
- 文件 >
- torchao.float8 >
- CastConfig
-
快捷方式
