我在Python中使用Tensorflow。 我想用LSTM单元创建一个RNN。 每个时间步长的输入看起来都像x = [1,2,3]。 我想将1馈入LSTM单元,将2馈入另一个,并将3馈入另一个。 然后,我想将lstm单元相互连接。 这样,我的输出将是x大小相同的矢量,其中第一个元素源自第一个单元格,依此类推。
据我所知tf.contrib.rnn.MultiRNNCell
确实堆叠了单元格,即单元格1的输出是单元格2的输入,依此类推。
同样,LSTM单元格中的num_units
参数具有使num_units
互连的单元格的效果,但是它们都获得相同的输入。
在我的示例中,将是每个单元格都获得整个向量x。
我错过了这种架构的包装器吗? 还是我必须自己写这个,而人们将如何去做呢?