ENH: add broadcast_shapes

Author: ev-br

array_api_strict/__init__.py

@@ -92,6 +92,11 @@
     "result_type",
 ]
 
+# TODO: only add for 2025.12
+from ._data_type_functions import broadcast_shapes
+__all__ += ["broadcast_shapes"]
+
+
 from ._dtypes import (
     int8,
     int16,

array_api_strict/_data_type_functions.py

@@ -62,6 +62,16 @@ def broadcast_arrays(*arrays: Array) -> list[Array]:
     ]
 
 
+def broadcast_shapes(*shapes: tuple[int, ...]) -> tuple[int, ...]:
+    """
+    Array API compatible wrapper for :py:func:`np.broadcast_shapes <numpy.broadcast_shapes>`.
+
+    See its docstring for more information.
+    """
+    # TODO: only define for 2025.12
+    return np.broadcast_shapes(*shapes)
+
+
 def broadcast_to(x: Array, /, shape: tuple[int, ...]) -> Array:
     """
     Array API compatible wrapper for :py:func:`np.broadcast_to <numpy.broadcast_to>`.