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.

参数:

*arrays (array) -- The input arrays.

返回:

The output arrays with the broadcasted shape.

返回类型:

tuple(array)