Closed shuguang101 closed 5 years ago
Hi, @andabi in master branch model.py line 54 your code is: out = out[..., -1] # (n, h)
is this want to take the last timestep output?
@shuguang101 actually it was fixed but not updated ;( It should be out = out[..., -1, :].
out = out[..., -1, :]
Hi, @andabi in master branch model.py line 54 your code is: out = out[..., -1] # (n, h)
is this want to take the last timestep output?