Imported from https://github.com/davidhalladay/CE-compressive-sensing-on-generative-model-pytorch