Imported from https://github.com/alecokas/lattice_rnn