Vq vae

shape