If it takes a tensor of shape (batch_size, seq_len, input_size), then it is fine. But when I open its code, it seems to accept the tensor of shape (seq_len, batch_size, input_size) which is a big problem. Can anyone confirm if I am right?