Sequential Attend, Infer, Repeat: Generative Modelling of Moving Objects
auto - encoding variational bayes .