Tensorflow在每个时间步将每个输入元素馈送到不同的LSTM单元

时间:2018-07-05 09:43:25

标签: python tensorflow input lstm rnn

我在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。

我错过了这种架构的包装器吗? 还是我必须自己写这个,而人们将如何去做呢?

0 个答案:

没有答案