Browse Source

fix lstm out size

tags/v0.3.0-alpha
zjun 5 years ago
parent
commit
181fd93e45
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/lstm.py

+ 1
- 1
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))

Loading…
Cancel
Save