keras lstm return sequence参数理解

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

lstm cell返回两组数据

  • hidden state返回的是根据当前input得到的output
  • cell state返回的是根据当前input以及长短时记忆获取的数据得到的output,这也是lstm的关键

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

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

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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容