mlx_rs::ops

Function eye_device

Source 
pub fn eye_device<T: ArrayElement>(
    n: i32,
    m: Option<i32>,
    k: Option<i32>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

See Array::eye