Skip to main content

array_split_n

Function array_split_n 

Source
pub fn array_split_n<T: Element>(
    a: &Array<T, IxDyn>,
    n: usize,
    axis: usize,
) -> FerrayResult<Vec<Array<T, IxDyn>>>
Expand description

Split an array into n sub-arrays along axis, allowing uneven sections.

Unlike split, this never errors on uneven division: the first axis_len % n sections have one extra element. This matches NumPy’s numpy.array_split(ary, n, axis) (integer-section variant).

§Errors

Returns FerrayError::AxisOutOfBounds if axis >= ndim. Returns FerrayError::InvalidValue if n == 0.