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)