model(x) will take a whole batch of tensors with the batch size in dim0.
E.g. if your model uses an nn.Linear module as the first layer, you might feed the complete dataset as [20000, nb_features] into your model.