Pytorch LSTM获取没有循环的输出最后状态

时间:2018-08-14 02:54:33

标签: lstm pytorch

我正在阅读Seq2Seq教程here。是否可以从上一步获得encoder_output而不使用此for循环?

for ei in range(input_length):
    encoder_output, encoder_hidden = encoder(
        input_tensor[ei], encoder_hidden)
    encoder_outputs[ei] = encoder_output[0, 0]

0 个答案:

没有答案