Imported from https://github.com/openai/pixel-cnn