Solved by suraj.pt in post #6 
              
              
                Let’s say your  batch_size=5 and seq_len=3. So each batch looks like 
batch = [
        [x1_1, x1_2, x1_3],
        [x2_1, x2_2, x2_3],
        ...
        [x5_1, x5_2, x5_3]
       ] # shape (batch_size, seq_len, input_size) 

where x{seq_id}_{timestep}

For a given timestep t, the RNN reads the t-…