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.UpDecoderBlock2D.UpDecoderBlock2D

next

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