fix categorical for softmax regression or in general for p with more than 1 dimension
fix categorical for softmax regression or in general for p with more than 1 dimension

wanderer2/pymc3,wanderer2/pymc3