Trait TensorQuantization 

pub trait TensorQuantization<T: Float> {
    // Required methods
    fn quantize_dynamic(
        &self,
        scheme: QuantizationScheme,
    ) -> RusTorchResult<QuantizedTensor<i8>>;
    fn quantize_static(
        &self,
        scale: f32,
        zero_point: i32,
    ) -> RusTorchResult<QuantizedTensor<i8>>;
    fn can_quantize(&self) -> bool;
}

Main quantization API for tensors テンソル用メイン量子化API

Required Methods

fn quantize_dynamic( &self, scheme: QuantizationScheme, ) -> RusTorchResult<QuantizedTensor<i8>>

Perform dynamic quantization 動的量子化を実行

fn quantize_static( &self, scale: f32, zero_point: i32, ) -> RusTorchResult<QuantizedTensor<i8>>

Perform static quantization with pre-computed parameters 事前計算されたパラメータでの静的量子化を実行

fn can_quantize(&self) -> bool

Check if tensor is suitable for quantization テンソルが量子化に適しているかチェック

Implementors

impl<T: Float + Quantizable<QuantizedType = i8>> TensorQuantization<T> for Tensor<T>