给出结论便于查询,具体分析过程见聊一聊PyTorch中LSTM的输出格式
- output保存了最后一层,每个time step的输出h,如果是双向LSTM,每个time step的输出h = [h正向, h逆向]
(同一个time step的正向和逆向的h连接起来)。 - h_n保存了每一层,最后一个time step的输出h,如果是双向LSTM,单独保存前向和后向的最后一个time step的输出h。
- c_n与h_n一致,只是它保存的是c的值。
给出结论便于查询,具体分析过程见聊一聊PyTorch中LSTM的输出格式