完全连接的层跟随LSTM层

时间:2017-01-10 06:42:43

标签: machine-learning tensorflow

如何组合Tensorflow完全连接层,然后是LSTM层。我的目标是提供batch_size = batch_size,序列长度seq_length和维度1的数据。目标是21 dim one_hot vector。

这是我试过的代码。它会抛出错误形状必须是等级2,但是对于W_first行是等级3。我做错了什么

data = tf.placeholder(tf.float32, [None, 20,1]) #Number of examples,number of input, dimension of each input
target = tf.placeholder(tf.float32, [None, 21])
W_first=tf.Variable(tf.random_normal([10,num_hidden]))
out_1= tf.matmul(data,W_first)
cell = tf.nn.rnn_cell.LSTMCell(num_hidden,state_is_tuple=True)
val, _ = tf.nn.dynamic_rnn(cell, out_1, dtype=tf.float32)

提前致谢!

0 个答案:

没有答案