美文网首页机器学习杂文
keras lstm return sequence参数理解

keras lstm return sequence参数理解

作者: yangy_fly | 来源:发表于2019-08-25 21:35 被阅读0次

    使用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结果一致

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

    相关文章

      网友评论

        本文标题:keras lstm return sequence参数理解

        本文链接:https://www.haomeiwen.com/subject/dlwkectx.html