如何在Keras或任何框架中的每个内存块中实现具有多个单元的LSTM层?

时间:2019-10-21 13:57:14

标签: tensorflow keras pytorch lstm block

我一直在寻找解决方案,以在Keras中实现每个存储块具有多个单元的LSTM层。 我使用PyTorch找到了一个关于同一问题的non-resolved post,但有关如何做却一无所获。

但是,在Hochreiter和Schmidhuber [1997]提出的原始LSTM模型中,每个LSTM块/单元可以包含几个单元:LSTM Network [Hochreiter, 1997]

有什么办法吗?

我目前正与Keras面临相同的问题,但是如果可以在PyTorch或其他任何工具上使用,我很乐意更改使用的框架。

我想用以下方法实现一个三层神经网络: -输入层 -具有LSTM单位的隐藏层:2个块,每个3个单元。 -输出层

任何框架都允许这样做吗?我确实需要实现2个块,每个块3个单元,而不是6个LSTM单元。

提前感谢您的时间,回答或建议。

0 个答案:

没有答案