beam search

[asmartsnail]

topk_decoder = TopKDecoder(decoder, 5) seq2seq = Seq2seq(encoder, topk_decoder)
When I use the above code in sample.py, I get the following error.

When I make the following changes in TopKDecoder.py, the code can run successfully, but the accuracy is only about 0.2.
Where is the problem?
metadata['sequence'] = [seq[0] for seq in p] =》metadata['sequence'] = [seq[:,0,:] for seq in p]