VQ-VAE: Looking Up the Code
Beginner
The problem, from first principles
After the nearest code is chosen, the quantized latent is simply that code vector. The decoder receives the code, so the index is all it needs to reconstruct the image.
From theory to code
Implement quantize(codebook, idx), which returns the code vectors at the given indices.
Constraints
- Works for one index or many.
Hints
Hint 1
Index the codebook with idx.
Keywords
research-papersunsuperviseddiscrete-latentsvae
Python
Run your code to see results
No results yet
Run your code to check it against the examples, or submit to run every test.