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

Functions

randn_tensor

This is a helper function that allows to create random tensors on the desired device with the desired dtype.

Classes

DiagonalGaussianDistribution

previous

modelzoo.vision.pytorch.dit.layers.vae.VAEModel.DecoderOutput

next

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