supplement another method
This commit is contained in:
@ -58,6 +58,11 @@ class RNN(nn.Module):
|
|||||||
# outs = outs.view(-1, TIME_STEP, 1)
|
# 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)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user