在dynamic_rnn中傳遞初始狀態

Question

我想用我自己的initial_state來使用方法tf.nn.dynamic_rnn。給予tf.nn.dynamic_rnn的單元格的狀態是一個tupel。我不明白的文檔：

如果cell.state_size是一個元組，這應該是一種在cell.state_size形狀[batch_size時，S]對於s張量的元組。

如何傳遞我的形狀（batchSize，hiddenSize）以便hiddenSize的元素是隱藏和單元狀態的初始狀態？

Answer 1

您可以使用init方法tf.contrib.rnn.LSTMStateTuple(c, h)生成LSTM狀態tupel，並將該tupel傳遞給tf.nn.dynamic_rnn以初始化狀態。