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