broadcast_shapes

array_api_extra.broadcast_shapes(*shapes, xp=None)

Compute the shape of the broadcasted arrays.

Deprecated since version 0.11.0: broadcast_shapes() is deprecated and will be removed in v1.0.0. array_api.broadcast_shapes() exists in the standard as of v2025.12.

Duplicates numpy.broadcast_shapes(), with additional support for None and NaN sizes.

Parameters:

  • *shapes (tuple[float | None, ...]) – Shapes of the arrays to broadcast.

  • xp (ModuleType | None) – The standard-compatible namespace to use for native delegation. Default: use the array-agnostic implementation.

Returns:

The shape of the broadcasted arrays.

Return type:

tuple[int | None, ...]

See also

numpy.broadcast_shapes

Equivalent NumPy function.

array_api.broadcast_arrays

Function to broadcast actual arrays.

Notes

This function accepts the Array API’s None for unknown sizes, as well as Dask’s non-standard math.nan. Regardless of input, the output always contains None for unknown sizes.

Examples

>>> import array_api_extra as xpx
>>> xpx.broadcast_shapes((2, 3), (2, 1))
(2, 3)
>>> xpx.broadcast_shapes((4, 2, 3), (2, 1), (1, 3))
(4, 2, 3)