modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKLOutput#
- class modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKLOutput[source]#
Bases:
object
Output of AutoencoderKL encoding method.
- Parameters
latent_dist (DiagonalGaussianDistribution) – Encoded outputs of Encoder represented as the mean and logvar of DiagonalGaussianDistribution. DiagonalGaussianDistribution allows for sampling latents from the distribution.
Methods
Attributes
latent_dist
- __init__(latent_dist: modelzoo.vision.pytorch.dit.layers.vae.utils.DiagonalGaussianDistribution) None #