在Tensorflow中生成特殊输出字后如何停止RNN?

问题描述 投票:3回答:1

我想实现一个编码器 - 解码器模型,用于序列到序列学习。

编码器逐字读取输入序列并更新其隐藏状态。

解码器使用编码器的隐藏状态来初始化其隐藏状态。然后生成关于最后生成的输出(y(t-1))及其隐藏状态的输出。我想在生成特殊输出()时停止此过程。实际上,我希望能够生成不同长度的输出。我怎么能在Tensorflow中做到这一点?

tensorflow recurrent-neural-network
1个回答
1
投票

我想你想要像sequence_lengthtf.nn.rnn。我也想要它,但似乎TensorFlow没有它。

我到目前为止所做的并找到了解决此限制的好方法是在列车时间使用EOS符号填充解码器标签。通常,您只需要其中一种,但填充其中很多都不会造成任何伤害。

在执行时,您可以手动控制每次迭代,以便在生成第一个EOS时停止,或者只运行预定义数量的时间步长,然后从输出中删除额外的EOS符号。解码器很快得知在第一个EOS之后,只有更多的EOS可以跟随。

© www.soinside.com 2019 - 2024. All rights reserved.