mlx_rs::ops

Function arange_device

Source 
pub fn arange_device<U, T>(
    start: impl Into<Option<U>>,
    stop: U,
    step: impl Into<Option<U>>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
where
    U: NumCast,
    T: ArrayElement,
Expand description

See Array::arange