LSTM(长短期记忆网络)中的关键参数包括输入门、遗忘门、输出门、细胞状态和隐藏状态。以下是如何进行推理计算的示例:
LSTM参数和公式
-
输入门(i_t) :决定输入的信息量。
-
遗忘门(f_t) :决定遗忘上一个状态的信息量。
-
细胞状态(C_t) :存储长期信息。
-
输出门(o_t) :决定输出的信息量。
-
隐藏状态(h_t) :输出短期记忆。
推理计算示例
假设我们有以下输入数据和参数:
计算步骤
总结:训练所得四个权重,然后不停的更新状态,遗忘并输出