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等）中的一個。但是，如果您保留了多個批次的狀態，因爲它的大小必須保持不變，所以無法工作。