ianDistribution

)


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.

FlaxAutoencoderKL
	

class d