Function as_strided

pub fn as_strided<'a>(
    a: impl AsRef<Array>,
    shape: impl IntoOption<&'a [i32]>,
    strides: impl IntoOption<&'a [usize]>,
    offset: impl Into<Option<usize>>,
) -> Result<Array>

Create a view into the array with the given shape and strides.

Example

use mlx_rs::{Array, ops::*};

let x = Array::from_iter(0..10, &[10]);
let y = as_strided(&x, &[3, 3], &[1, 1], 0);