diff --git a/mindspore/nn/layer/lstm.py b/mindspore/nn/layer/lstm.py index 84c156a1c2..bdc49739ac 100755 --- 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))