mlx.core.broadcast_arrays

broadcast_arrays(*arrays: array, stream: None | Stream | Device = None) → Tuple[array, ...]

Broadcast arrays against one another.

The broadcasting semantics are the same as Numpy.

Parameters:

*arrays (array) – The input arrays.

Returns:

The output arrays with the broadcasted shape.

Return type:

tuple(array)