I guess that means that some of the training examples have more dimensions than others. The DataLoader assumes that each example has the same number of features, because that’s how it’s concatenated into a batch.