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

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

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

如何使用

output

、

hn

和

cn

提取最后的前向和后向隐藏状态

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