ENH: add broadcast_shapes if array_api_version is at least 2025.12

Author: ev-br

array_api_strict/__init__.py

@@ -92,6 +92,7 @@
     "result_type",
 ]
 
+
 from ._dtypes import (
     int8,
     int16,
@@ -335,6 +336,11 @@
     '__version__',
 ]
 
+if get_array_api_strict_flags()['api_version'] >= '2025.12':
+    from ._data_type_functions import broadcast_shapes
+    __all__ += ["broadcast_shapes"]
+
+
 try:
     from ._version import __version__  # type: ignore[import-not-found,unused-ignore]
 except ImportError:

array_api_strict/_data_type_functions.py

@@ -62,6 +62,15 @@ 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.
+    """
+    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>`.