The input to nn.GRU should be [seq_len, batch_size, input_size]. In your case, you would have to add the batch dimension to your data. I assume the features are supposed to represent the “time dimension”.