From 181fd93e45d1c728206ac54ab8a856b14c8cb71b Mon Sep 17 00:00:00 2001
From: zjun <zhangjun0@huawei.com>
Date: Thu, 23 Apr 2020 15:26:30 +0800
Subject: [PATCH] fix lstm out size

---
 mindspore/nn/layer/lstm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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