标签: machine-learning deep-learning pytorch lstm sequence-to-sequence
我不确定如何在Pytorch的双向LSTM中选择最后的隐藏/单元状态。
output, (hn, cn) = bi_lstm(input, (h0, c0))
如何使用output,hn和cn来提取最后的前向和后向隐藏状态?
output
hn
cn
对于后向LSTM,我想提取在向后处理整个序列后得到的隐藏状态。