Because it will be used as a parameter of the next shuffle.
I find that reshape can be done in the next step when calculating weights_per_label_and_feature in the _naive_bayes_mapper