Pytorch版本实现的基于MNIST的VQVAE(仅供学习)
VQ-VAE是一种改进的VAE,专门用于学习离散的潜在表示空间。VQ-VAE通过引入向量量化(vector quantization)技术，将连续的潜在向量映射到离散的嵌入空间。向量量化指的是输入在经过编码器后, 会被映射到一个离散的码本(codebook), 码本由一组离散的嵌入向量组成。VQ-VAE通过将每个连续特征向量与码本中的向量进行比较,选择最接近的嵌入向量(最近邻),这就是量化过程。
- VQVAE-architecture
