Struct SplitResidualVectorQuantizer

pub struct SplitResidualVectorQuantizer { /* private fields */ }

Implementations

impl SplitResidualVectorQuantizer

pub fn new( dim: usize, input_dim: Option<usize>, output_dim: Option<usize>, n_q: usize, bins: 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 SplitResidualVectorQuantizer

fn clone(&self) -> SplitResidualVectorQuantizer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SplitResidualVectorQuantizer

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

Formats the value using the given formatter.