Commit 61d6dc6

authored

supplement another method

1 parent 2748b02 commit 61d6dc6Copy full SHA for 61d6dc6

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

Lines changed: 5 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,11 @@ def forward(self, x, h_state):`
`57`	`57`	`# outs = self.out(r_out)`
`58`	`58`	`# outs = outs.view(-1, TIME_STEP, 1)`
`59`	`59`	`# return outs, h_state`
	`60`	`+`
	`61`	`+ # or even simpler, since nn.Linear can accept inputs of any dimension`
	`62`	`+ # and returns outputs with same dimension except for the last`
	`63`	`+ # outs = self.out(r_out)`
	`64`	`+ # return outs`
`60`	`65`
`61`	`66`	`rnn = RNN()`
`62`	`67`	`print(rnn)`

Comments

(0)