mlx_rs::ops

Function broadcast_arrays_device

Source
pub fn broadcast_arrays_device(
    arrays: &[impl AsRef<Array>],
    stream: impl AsRef<Stream>,
) -> Result<Vec<Array>>
Expand description

Broadcast a vector of arrays against one another. Returns an error if the shapes are broadcastable.

ยงParams

  • arrays: The arrays to broadcast.