Struct VqCodebook 

pub struct VqCodebook {
    pub embeddings: Vec<f32>,
    pub n_codes: usize,
    pub code_dim: usize,
    pub ema_momentum: f32,
    pub commitment_weight: f32,
    pub ema_counts: Vec<f32>,
    pub ema_sum: Vec<f32>,
}

EMA-maintained vector-quantisation codebook.

Stores K code vectors of dimension C as a flat [K × C] row-major buffer. EMA running statistics (ema_counts, ema_sum) are used for the Laplace-smoothed online codebook update rule:

    n_k  ← m·n_k + (1-m)·|S_k|
    e_k  ← m·e_k_sum + (1-m)·Σ_{z∈S_k} z   (running sum)
    code_k ← e_k_sum / n_k                   (normalised)

Fields

embeddings: Vec<f32>

Codebook entries E ∈ ℝ^{K×C}, row-major. Length = K * C.

n_codes: usize

Number of codes K.

code_dim: usize

Code dimensionality C.

ema_momentum: f32

EMA momentum m for codebook update (closer to 1 → slower update).

commitment_weight: f32

Commitment loss weight β.

ema_counts: Vec<f32>

EMA usage counts per code [K]. Initialised to 1 (Laplace smoothing).

ema_sum: Vec<f32>

EMA running sums per code [K × C]. Initialised to the code vectors.

Trait Implementations

impl Clone for VqCodebook

fn clone(&self) -> VqCodebook

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for VqCodebook

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

Formats the value using the given formatter.