Я не уверен, как выбрать последние скрытые состояния / ячейки в двунаправленном LSTM в Pytorch.
output, (hn, cn) = bi_lstm(input, (h0, c0))
Как я могу использовать output
, hn
и cn
, чтобы извлечь последние скрытые состояния вперед и назад?
В случае обратного LSTM, Iхочу извлечь скрытое состояние, которое я получаю после обработки всей последовательности в обратном направлении.