Skip to content

Commit 2748b02

Browse files
fill up the missing view() in forward() in comment
1 parent 40bcdb5 commit 2748b02

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

tutorial-contents/403_RNN_regressor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def forward(self, x, h_state):
5555
# instead, for simplicity, you can replace above codes by follows
5656
# r_out = r_out.view(-1, 32)
5757
# outs = self.out(r_out)
58+
# outs = outs.view(-1, TIME_STEP, 1)
5859
# return outs, h_state
5960

6061
rnn = RNN()

0 commit comments

Comments
 (0)