码迷,mamicode.com
首页 > 其他好文 > 详细

tf.nn.rnn_cell.MultiRNNCell

时间:2018-11-15 22:42:23      阅读:327      评论:0      收藏:0      [点我收藏+]

标签:neu   oat   type   put   float   layer   相同   一个   out   

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

标签:neu   oat   type   put   float   layer   相同   一个   out   

原文地址:https://www.cnblogs.com/yaos/p/9966081.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!