Struct VectorQuantization

pub struct VectorQuantization { /* private fields */ }

Implementations

impl VectorQuantization

pub fn new( dim: usize, codebook_size: usize, codebook_dim: Option<usize>, vb: VarBuilder<'_>, ) -> Result<Self>

pub fn encode(&self, xs: &Tensor) -> Result<Tensor>

pub fn decode(&self, codes: &Tensor) -> Result<Tensor>

Trait Implementations

impl Clone for VectorQuantization

fn clone(&self) -> VectorQuantization

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for VectorQuantization

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

Formats the value using the given formatter.