Function mlx_roll_axis 

pub unsafe extern "C" fn mlx_roll_axis(
    res: *mut mlx_array,
    a: mlx_array,
    shift: *const c_int,
    shift_num: usize,
    axis: c_int,
    s: mlx_stream,
) -> c_int