machine-learning - 如何在 Pytorch 中获取双向 LSTM 的最后隐藏状态？

Question

我不确定如何在 Pytorch 的双向 LSTM 中选择最后的隐藏/单元状态。

output, (hn, cn) = bi_lstm(input, (h0, c0))

我如何使用 output, hn来cn提取最后的前向和后向隐藏状态？

在反向 LSTM 的情况下，我想提取在反向处理整个序列后得到的隐藏状态。

标签： machine-learningdeep-learningpytorchlstmsequence-to-sequence