diffusion_models.models.vae.VariationalAutoencoder

class diffusion_models.models.vae.VariationalAutoencoder(in_channels=3, hidden_dim=256)[source]

Bases: Module

Class implementing a Variational Autoencoder.

__init__(in_channels=3, hidden_dim=256)[source]

Methods

__init__([in_channels, hidden_dim])

forward(x)

reparameterize(mu, sigma)