andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #58 from keineahnung2345/403-view

Browse Source 
403_RNN_regressor.py missing line in comment
This commit is contained in:
Morvan
2018-11-13 10:22:41 +08:00
committed by GitHub
parent 40bcdb5c6c 61d6dc6757
commit 903820ef35
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tutorial-contents/403_RNN_regressor.py
View File

@ -55,8 +55,14 @@ class RNN(nn.Module):
# instead, for simplicity, you can replace above codes by follows # instead, for simplicity, you can replace above codes by follows
# r_out = r_out.view(-1, 32) # r_out = r_out.view(-1, 32)
# outs = self.out(r_out) # outs = self.out(r_out)
# outs = outs.view(-1, TIME_STEP, 1)
# return outs, h_state # 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() rnn = RNN()
print(rnn) print(rnn)