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

class modelzoo.vision.pytorch.dit.layers.vae.VAEModel.DecoderOutput[source]#

Bases: object

Output of decoding method.

Parameters: sample (torch.FloatTensor of shape (batch_size, num_channels, height, width)) – Decoded output sample of the model. Output of the last layer of the model.

Methods

Attributes

sample

__init__(sample: torch.FloatTensor) → None#

previous

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

next

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