Tensorflow RNN示例限于固定批量大小？

Question

在Tensorflow上查看RNN example时，如何构造初始状态存在问题。在构建图表时，我们将图表限制为仅处理一个批量大小的输入。这对我来说是一个问题，因为我希望能够提供一个示例并获得单个示例的预测。

代码的制约，这是该部分：

initial_state = state = tf.zeros([batch_size, lstm.state_size]) 

所以我的问题是如何展开的例子，这样我可以用一个变量批量大小，这样我可以使用培训同型号与批量大小，然后使用单个示例进行预测？

Answer 1

这就是我如何做到这一点。你能通过batch_size作为像这样的变量：

batch_size = tf.placeholder(tf.int32) 
init_state = cell.zero_state(batch_size, tf.float32) 

其中cell是RNN细胞（BasicLSTMCell，BasicGRUCell，MultiRNNCell等）中的一个。但是，如果您保留了多个批次的状态，因为它的大小必须保持不变，所以无法工作。