具有多个输出的tensorflow seq2seq

Question

我刚开始在tensorflow上工作，时间不长。我正在研究seq2seq模型并使用seq2seq示例代码。

我想修改seq2seq模型代码，以获得强化学习模型的top-k输出(k是5或10)，而不是top-1输出。

首先，我想我应该以某种方式修改seq2seq的解码器部分，但我不知道该修改哪部分。

有没有针对这个问题的参考或代码？

Answer 1

查看https://github.com/tensorflow/tensorflow/issues/654。关于这一点有一些讨论，但还没有有效的例子。

Answer 2

tf.contrib.seq2seq.BeamSearchDecoder会为你施展魔法的。