Struct AutoEncoderKL

pub struct AutoEncoderKL {
    pub config: AutoEncoderKLConfig,
    /* private fields */
}

Fields

config: AutoEncoderKLConfig

Implementations

impl AutoEncoderKL

pub fn new( vs: VarBuilder<'_>, in_channels: usize, out_channels: usize, config: AutoEncoderKLConfig, ) -> Result<Self>

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

Returns the distribution in the latent space.

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

Takes as input some sampled values.

Trait Implementations

impl Debug for AutoEncoderKL

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

Formats the value using the given formatter.