写出 VQ-VAE 的前向与训练伪代码。