Struct Observer 

pub struct Observer { /* private fields */ }

Observer for tracking tensor statistics during calibration

Implementations

impl Observer

pub fn new(observer_type: ObserverType) -> Self

Create a new observer

pub fn new_histogram(num_bins: usize) -> Self

Create a new histogram observer with specified number of bins

pub fn new_percentile(percentile: f32) -> Self

Create a new percentile observer with specified percentile

pub fn update(&mut self, tensor: &Tensor) -> TorshResult<()>

Update observer with new tensor (optimized with parallel processing)

pub fn calculate_qparams(&self, dtype: DType) -> TorshResult<(f32, i32)>

Calculate quantization parameters from observed statistics

pub fn detect_outliers(&self, data: &[f32], factor: f32) -> (Vec<f32>, usize)

Detect and remove outliers using IQR method

pub fn get_statistics(&self) -> HashMap<String, f32>

Get comprehensive statistics from the observer

pub fn observer_type(&self) -> ObserverType

Get the observer type

pub fn get_min_max(&self) -> (f32, f32)

Get the current min/max values

pub fn num_batches(&self) -> usize

Get number of processed batches

pub fn reset(&mut self)

Reset the observer state

impl Observer

Factory functions for creating observers

pub fn min_max() -> Self

Create a MinMax observer

pub fn moving_average() -> Self

Create a MovingAverage observer

pub fn histogram() -> Self

Create a Histogram observer with default bins

pub fn histogram_with_bins(num_bins: usize) -> Self

Create a Histogram observer with custom number of bins

pub fn percentile() -> Self

Create a Percentile observer with default percentile

pub fn percentile_with_value(percentile: f32) -> Self

Create a Percentile observer with custom percentile

Trait Implementations

impl Debug for Observer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.