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

!2057 Lstm example fix 

Merge pull request !2057 from chenweifeng/lstm
This commit is contained in:
mindspore-ci-bot 2020-06-12 18:30:39 +08:00 committed by Gitee
parent 30b19165a2 c4da612ee8
commit af3488c000
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/model_zoo/lstm.py
View File

@ -88,6 +88,6 @@ class SentimentNet(nn.Cell):
embeddings = self.trans(embeddings, self.perm)
output, _ = self.encoder(embeddings, (self.h, self.c))
# states[i] size(64,200) -> encoding.size(64,400)
encoding = self.concat((output[0], output[1]))
encoding = self.concat((output[0], output[-1]))
outputs = self.decoder(encoding)
return outputs