Struct Vae 

pub struct Vae { /* private fields */ }

Variational Autoencoder for encoding/decoding between pixel and latent space.

Implementations

impl Vae

pub fn new( vs: VarBuilder<'_>, latent_channels: usize, scaling_factor: f64, ) -> Result<Self>

Load a VAE from a VarBuilder.

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

Encode an image to latent space (returns the mean of the posterior).

• image: (B, 3, H, W) tensor in [-1, 1] range.

Returns (B, latent_channels, H/8, W/8).

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

Decode a latent tensor back to pixel space.

• latents: (B, latent_channels, h, w) scaled latent tensor.

Returns (B, 3, H, W) in [-1, 1] range.

Trait Implementations

impl Debug for Vae

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

Formats the value using the given formatter.