pytorch中lstm输出问题

给出结论便于查询,具体分析过程见聊一聊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的值。