Imported from https://github.com/chandlersupple/Tensorflow-PixelCNN