Closed Yuhuishishishi closed 4 years ago
The shape for the hidden and cell state tensor should be [num_layers(=1) * num_directions(=2), batch_size, n_hidden]
I plan to upgrade the code a little further according to the corresponding below PR:
The shape for the hidden and cell state tensor should be [num_layers(=1) * num_directions(=2), batch_size, n_hidden]