回答編集履歴
2
誤字の修正
answer
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
NStepLSTMの使い方を間違えていました。
|
|
2
|
-
`
|
|
2
|
+
`ys[-1]`は、最終ステップの出力ではなく、ミニバッチ内の1つのデータに対する全ステップ出力を指しているようです。
|
|
3
3
|
|
|
4
4
|
なので、正しく最終ステップの出力を得るには以下の様にリスト内の各要素の末尾を取って連結すれば良いみたいです。
|
|
5
5
|
```python
|
1
誤字の修正
answer
CHANGED
|
@@ -4,7 +4,7 @@
|
|
|
4
4
|
なので、正しく最終ステップの出力を得るには以下の様にリスト内の各要素の末尾を取って連結すれば良いみたいです。
|
|
5
5
|
```python
|
|
6
6
|
def __call__(self, x):
|
|
7
|
-
hy, cy, ys = self.lstm(None, None,
|
|
7
|
+
hy, cy, ys = self.lstm(None, None, x)
|
|
8
8
|
h = F.vstack(map(lambda y: y[-1], ys))
|
|
9
9
|
y = self.linear(h)
|
|
10
10
|
return y
|