p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!608 Fix LSTM output size 

Merge pull request !608 from zjun/fix_lstm_output
This commit is contained in:
mindspore-ci-bot 2020-04-27 14:48:56 +08:00 committed by Gitee
parent 68f6064931 181fd93e45
commit 417753e9b3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/nn/layer/lstm.py
View File

@ -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))