!2057 Lstm example fix

Merge pull request !2057 from chenweifeng/lstm
5 years ago · af3488c000
parent 30b19165a2 c4da612ee8
commit af3488c000
1 changed files with 1 additions and 1 deletions
--- a/mindspore/model_zoo/lstm.py
+++ b/mindspore/model_zoo/lstm.py
@ -88,6 +88,6 @@ class SentimentNet(nn.Cell):
        embeddings = self.trans(embeddings, self.perm)
        output, _ = self.encoder(embeddings, (self.h, self.c))
        # states[i] size(64,200)  -> encoding.size(64,400)
-        encoding = self.concat((output[0], output[1]))
+        encoding = self.concat((output[0], output[-1]))
        outputs = self.decoder(encoding)
        return outputs