This is stored in lstm_oupt[-1], where the -1 index means "last cell" in this context. The output could also have been fetched as h_n directly. The output of the LSTM layer has log-softmax() ...