机器学习杂文

keras lstm return sequence参数理解

2019-08-25  本文已影响0人  yangy_fly

使用keras构建多层lstm网络时,除了最后一层lstm,中间过程的lstm中的return sequence参数需要设置为true。

lstm cell返回两组数据

return sequence为true,则返回所有hidden state的结果,当构建多层lstm时,上下层全连接需要该结果,因此需要置为true。

当lstm为单层或者最后一层时,return sequence置为false,则此时返回的结果与cell state结果一致

*** 若有不对之处,希望共同交流,指出错误,谢谢 ***

上一篇 下一篇

猜你喜欢

热点阅读