You probably need to transpose the categorical observations. Batched dimensions have to be on the left. I assume you have 3 pairs of 2 categorical variables?
If your Dirichlet prior varied across observations, it would have to have shape (3, 2, 2), in case that helps