pytorch 学习 | 使用pytorch动手实现LSTM模块

LSTM 简介 LSTM是RNN中一个较为流行的网络模块。主要包括输入,输入门,输出门,遗忘门,激活函数,全连接层(Cell)和输出。其结构如下: 上述公式不做解释,我们只要大概记得以下几个点就可以了: 当前时刻LSTM模块的输入有来自当前时刻的输入值,上一时刻的输出值,输入值和隐含层输出值,就是一共有四个输入值,这意味着一个LSTM模块的输入量是原来普通全连接层的四倍左右,计算量多了许多。 所谓
相关文章
相关标签/搜索