jax-js
    Preparing search index...

    Function split

    • Split an array into multiple sub-arrays along an axis.

      Parameters

      • a: ArrayLike

        The input array to split.

      • indicesOrSections: number | number[]

        If an integer, it indicates the number of equal sections to create along the specified axis. If a list of integers, it specifies the indices at which to split the array.

      • axis: number = 0

        The axis along which to split the array. Default is 0.

      Returns Array[]