Marla:

For an RNN, ‘output’ is given as (seq_len, batch, num_directions * hidden_size) with ‘batch’ no longer represented (as per the documentation).


What do you mean the batch is no longer represented? It’s in the second dimension :). If you need it in the first one you could simply do output.transpose(0,1)