Merge pull request #58 from keineahnung2345/403-view
403_RNN_regressor.py missing line in comment
This commit is contained in:
@ -55,8 +55,14 @@ 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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user