快捷方式

MarlinQQQTensor

class torchao.dtypes.MarlinQQQTensor(tensor_impl: AQTTensorImpl, block_size: Tuple[int, ...], shape: Size, quant_min: Optional[Union[int, float]] = None, quant_max: Optional[Union[int, float]] = None, zero_point_domain: ZeroPointDomain = ZeroPointDomain.INT, dtype=None, strides=None)[source]

MarlinQQQ 量化張量子類,繼承自 AffineQuantizedTensor 類。

要了解在 choose_qparams_and_quantize_affine_qqq、marllin qqq 量化的量化和反量化過程中發生的情況,請訪問 https://github.com/pytorch/ao/blob/main/torchao/quantization/quant_primitives.py 並檢視兩個量化原始操作:choose_qparams_and_quantize_affine_qqq 和 dequantize_affine_qqq

dequantize() Tensor[source]

給定一個量化張量,對其進行反量化並返回反量化後的浮點張量。

classmethod from_hp_to_intx(input_float: Tensor, block_size: Tuple[int, ...], quant_min: Optional[int] = None, quant_max: Optional[int] = None, zero_point_domain: ZeroPointDomain = ZeroPointDomain.INT, _layout: Optional[Layout] = None)[source]

將浮點張量轉換為 Marlin QQQ 量化張量。

文件

訪問 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源