Concrete Dropout
auto - encoding variational bayes .