Fix missing num_var parameter in suff_stat reshape

[braun-steven]

When the input for CategoricalArray was len(x.shape)==3, the reshape operation for the stats variable was missing the num_var variable and resulted in a wrong output shape (i.e. the -1 as first reshape-argument would then merge the batch size and num_var into a single dimension).