Tensorflow RNNの例は固定バッチサイズに制限されていますか？

Question

Tensorflow imでRNN exampleを見て、初期状態の構成に問題がある。グラフの作成時に、グラフは1つのバッチサイズの入力のみを処理するように制限されています。 1つの例でフィードを作成し、その1つの例の予測を取得したいので、これは私の問題です。

これを制限したコードの一部である：

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

だから私の質問は、私は訓練のために同じモデルを使用できるように、私は変数バッチサイズを使用できるように、どのように私は例を拡張することができますですバッチサイズを使用して、単一の例を予測に使用しますか？

Answer 1

これは私がこれをやっている方法です。 cellはRNN細胞（BasicLSTMCell、BasicGRUCell、MultiRNNCell、など）の一つである

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

：あなたはこのような変数としてbatch_sizeを渡すことができます。しかし、サイズが一定でなければならないので、複数のバッチで状態を保存している場合は機能しません。