Imported from https://github.com/casey-barr/generative-models-in-tensorflow