fix list-of-pair transposition in `dot_general` test under py3
fix list-of-pair transposition in `dot_general` test under py3

google/jax,google/jax,google/jax,tensorflow/probability,google/jax,tensorflow/probability