supplement another method

2018-11-12 14:24:09 +08:00
parent 2748b02bd6
commit 61d6dc6757
1 changed files with 5 additions and 0 deletions
--- a/tutorial-contents/403_RNN_regressor.py
+++ b/tutorial-contents/403_RNN_regressor.py
@ -57,6 +57,11 @@ class RNN(nn.Module):
        # outs = self.out(r_out)
        # outs = outs.view(-1, TIME_STEP, 1)
        # return outs, h_state
+        
+        # or even simpler, since nn.Linear can accept inputs of any dimension 
+        # and returns outputs with same dimension except for the last
+        # outs = self.out(r_out)
+        # return outs

 rnn = RNN()
 print(rnn)