FloatTensor

)


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.



Output of decoding method.

VQEncoderOutput
	

class diffusers.models.vq_model.VQEncoderOutput

<
source
>
(
latents: FloatTensor

)


Parameters 

latents (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —
Encoded output sample of the mo