PyTorch Beam Search Decoding

PyTorch-Beam-Search

基于 https://github.com/shawnwun/NNDIAL 的 seq2seq 模型的束搜索解码的 PyTorch 实现。对每个句子单独进行解码，并将节点存储在优先队列中。

用法：可以通过BeamSearchNode.eval指定额外的解码奖励。适用于有注意力和无注意力的模型。