Imported from https://github.com/ardapekis/pixel-rnn