mlx_rs::ops

Function diag

Source
pub fn diag(a: impl AsRef<Array>, k: impl Into<Option<i32>>) -> Result<Array>
Expand description