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#