如何使用带有张量流的while_loop在每个步骤中保存输出？

Question

我想构建一个具有数千次步长的RNN，所以正确的方法是使用while_loop函数，因为GPU将在for循环中内存不足。

但是我找不到在每一步中保存rnn输出的方法。我尝试使用全局列表或使用tf.concat（）来累积输出。都没有奏效。似乎while_loop（）只能用于获取最终输出。

是否有任何解决方案可以获得所有输出？

Answer 1

尝试tf.nn.dynamic_rnn使用while_loop和TensorArray对象完成此操作。