tf.nn.rnn_cell.MultiRNNCell

Posted yaos

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tf.nn.rnn_cell.MultiRNNCell相关的知识,希望对你有一定的参考价值。

num_units = [200, 100]
cells = [tf.nn.rnn_cell.GRUCell(num_units=n) for n in num_units]
stacked_rnn_cell = tf.nn.rnn_cell.MultiRNNCell(cells)
rnn_outputs, states = tf.nn.dynamic_rnn(stacked_rnn_cell, X, dtype=tf.float32)

最后一个cellnum_units必须和下面的n_neurons相同

stacked_rnn_outputs = tf.reshape(rnn_outputs, [-1, n_neurons])
stacked_outputs = tf.layers.dense(stacked_rnn_outputs, n_outputs)
outputs = tf.reshape(stacked_outputs, [-1, n_steps, n_outputs])

以上是关于tf.nn.rnn_cell.MultiRNNCell的主要内容,如果未能解决你的问题,请参考以下文章