标签： python deep-learning pytorch machine-translation opennmt
我正在使用OpenNMT-py框架，遇到了令人费解的事情。我似乎只能从seq2seq LSTM模型中提取隐藏状态，而不是单元状态。 onmt/encoders/rnn_encoder.py中的编码器类返回enc_state, memory_bank, lengths。隐藏状态在memory_bank中，大小为[sentence_length,batch_size,hidden_dim].，因此单元格状态在哪里？
onmt/encoders/rnn_encoder.py
enc_state, memory_bank, lengths
memory_bank
[sentence_length,batch_size,hidden_dim].