如何在Pytorch中采取双向LSTM的最后隐藏状态?

时间:2019-05-30 11:40:48

标签: machine-learning deep-learning pytorch lstm sequence-to-sequence

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

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

如何使用outputhncn来提取最后的前向和后向隐藏状态?

对于后向LSTM,我想提取在向后处理整个序列后得到的隐藏状态。

0 个答案:

没有答案