[NumPy] Fix uses of deprecated multidimensional NumPy indexing with a non-tuple index.

NumPy 1.23 removes support for non-tuple indexing of NumPy arrays (https://numpy.org/devdocs/release/1.23.0-notes.html#expired-deprecations). The workaround is to convert multidimensional indices to a tuple.

PiperOrigin-RevId: 465319674

Author: hawkinsp

tensorflow_probability/python/internal/backend/numpy/numpy_math.py

@@ -223,7 +223,7 @@ def _confusion_matrix(
   if weights is None:
     weights = 1
   if not JAX_MODE:
-    np.add.at(cmatrix, [labels, predictions], weights)
+    np.add.at(cmatrix, (labels, predictions), weights)
     return cmatrix
   return cmatrix.at[(labels, predictions)].add(weights)