!608 Fix LSTM output size

Merge pull request !608 from zjun/fix_lstm_output
5 years ago · 417753e9b3
parent 68f6064931 181fd93e45
commit 417753e9b3
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/layer/lstm.py
+++ b/mindspore/nn/layer/lstm.py
@ -149,7 +149,7 @@ class LSTM(Cell):
        if self.batch_first:
            x = self.transpose1(x, (1, 0, 2))
        h0, c0 = hx
-        output, hn, cn, _ = self.lstm(x, h0, c0, self.weight)
+        output, hn, cn, _, _ = self.lstm(x, h0, c0, self.weight)
        if self.batch_first:
            output = self.transpose2(output, (1, 0, 2))
        return (output, (hn, cn))