


VQ-VAE model from the paper Neural Discrete Representation Learning by Aaron van den Oord, Oriol Vinyals and Koray
Kavukcuoglu.
This model inherits from ModelMixin. Check the superclass documentation for the generic methods the library
implements for all the model (such as downloading or saving, etc.)

forward

<
source
>
(
sample: FloatTensor
return_dict: bool = True

)


Parameters 

sample (torch.FloatTensor) — Input sample.


return_dict