fill up the missing view() in forward() in comment

This commit is contained in:
keineahnung2345
2018-11-12 14:18:49 +08:00
committed by GitHub
parent 40bcdb5c6c
commit 2748b02bd6

View File

@ -55,6 +55,7 @@ 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
rnn = RNN()