ussianDistribution  )   Parameters   latent_dist (FlaxDiagonalGaussianDistribution) —
Encoded outputs of Encoder represented as the mean and logvar of FlaxDiagonalGaussianDistribution.
FlaxDiagonalGaussianDistribution allows for sampling latents from the distribution.    Output of AutoencoderKL encoding method.  replace  < source > ( **updates  )    “Returns a new object replacing the specified fields with new values.  FlaxDecoderOutput  class 