Commit 2748b02

authored

fill up the missing view() in forward() in comment

1 parent 40bcdb5 commit 2748b02Copy full SHA for 2748b02

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -55,6 +55,7 @@ def forward(self, x, h_state):`
`55`	`55`	`# instead, for simplicity, you can replace above codes by follows`
`56`	`56`	`# r_out = r_out.view(-1, 32)`
`57`	`57`	`# outs = self.out(r_out)`
	`58`	`+ # outs = outs.view(-1, TIME_STEP, 1)`
`58`	`59`	`# return outs, h_state`
`59`	`60`
`60`	`61`	`rnn = RNN()`

Comments

(0)