I think RNN model in Pytorch is batch major, the shape of input data should be (  n_batch, n_step, input_size). But in the demo code about the word_language_model https://github.com/pytorch/examples/blob/master/word_language_model/main.py 6 , it seems that the data input to rnn model is time major, which shape is ( n_step, n_batch, input_size). This confused me. Could anyone explain it? Thanks