This would thus be the number of samples in the batch, wouldn’t it?
If so, then you should not flatten this dimension with the features, but could use the input directly as [batch_size=171, nb_features=11] and set in_features=11 in the linear layer.