Imported from https://github.com/sherjilozair/char-rnn-tensorflow