Wasserstein Auto-Encoders
auto - encoding variational bayes .