mlx_rs::ops

Function repeat_axis_device

Source 
pub fn repeat_axis_device<T: ArrayElement>(
    array: Array,
    count: i32,
    axis: i32,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

See Array::repeat