mlx.core.broadcast_arrays

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

相互广播数组。

广播语义与 Numpy 相同。

参数:

*arrays (数组) – 输入数组。

返回:

具有广播形状的输出数组。

返回类型:

tuple(数组)