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

Question

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

但我找不到在每个步骤中保存 rnn 输出的方法。我尝试使用全局列表或使用 tf.concat() 来累积输出。都没有奏效。好像 while_loop() 只能用来获取最终输出。

有没有办法得到所有的输出？

Answer 1

试试tf.nn.dynamic_rnn，它使用while_loop 和TensorArray 对象来做到这一点。