使用tf.nn.dynamic_rnn生成多个隐藏层的{LSTM RNN}

时间:2017-09-07 03:19:52

标签: python tensorflow deep-learning tensorflow-serving

我阅读了tf.dynamic.rnn的文档 https://www.tensorflow.org/api_docs/python/tf/nn/dynamic_rnn

并用它来制作多个时间步骤的单层rnn。我想知道我是否可以使用tf.dynamic.rnn来堆叠多个隐藏层。有可能这样做吗?

1 个答案:

答案 0 :(得分:2)

请查看以下代码段:

Layers = 2

LTSM_Layer = tf.contrib.rnn.BasicLSTMCell(rnn_size)
Cell = tf.contrib.rnn.MultiRNNCell([LTSM_Layer] * Layers)

然后将此单元格传递给tf.dynamic.rnn