Merge pull request #58 from keineahnung2345/403-view

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

View File

@ -55,7 +55,13 @@ class RNN(nn.Module):
# instead, for simplicity, you can replace above codes by follows
# r_out = r_out.view(-1, 32)
# 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)