PyTorch Beam Search Decoding

PyTorch-Beam-Search

基於 https://github.com/shawnwun/NNDIAL 的 seq2seq 模型的束搜尋解碼的 PyTorch 實作。對每個句子單獨進行解碼，並將節點儲存在優先隊列中。

用法：可以透過BeamSearchNode.eval指定額外的解碼獎勵。適用於有註意力和無注意力的模型。