Vector Quantized Wasserstein Auto-Encoder