modelzoo.vision.pytorch.dit.layers.vae.utils.DiagonalGaussianDistribution

class modelzoo.vision.pytorch.dit.layers.vae.utils.DiagonalGaussianDistribution

Bases: object

Methods

kl
mode
nll
sample

__init__(parameters, deterministic=False)

previous

modelzoo.vision.pytorch.dit.layers.vae.utils.randn_tensor

next

modelzoo.vision.pytorch.dit.model