用输出门(output gate)来控制单元状态 有多少输出到LSTM的当前输出值 。
我们先来看一下遗忘门:
上式中, 是遗忘门的权重矩阵, 表示把两个向量连接成一个更长的向量, 是遗忘门的偏置项, 是
sigmoid函数。如果输入的维度是 ,隐藏层的维度是 ,单元状态的维度是 (通常 ),则遗忘门的权重矩
阵 维度是 。事实上,权重矩阵 都是两个矩阵拼接而成的:一个是 ,它对应着输入项 ,
其维度为 ;一个是 ,它对应着输入项 ,其维度为 。 可以写为:
下图显示了遗忘门的计算:
接下来看看输入门:
上式中, 是输入门的权重矩阵, 是输入门的偏置项。下图表示了输入门的计算:
接下来,我们计算用于描述当前输入的单元状态 ,它是根据上一次的输出和本次输入来计算的:
下图是 的计算: