§
    `ƒiÉ  ã                   ó<   — d dl Z d dlmZ dd„Zd„ Zd„ Zdd„Zd„ ZdS )	é    N)Ú_corec                 ó.   — t          j        | ||¦  «        S )aD  Splits an array into multiple sub arrays along a given axis.

    This function is almost equivalent to :func:`cupy.split`. The only
    difference is that this function allows an integer sections that does not
    evenly divide the axis.

    .. seealso:: :func:`cupy.split` for more detail, :func:`numpy.array_split`

    )r   Úarray_split)ÚaryÚindices_or_sectionsÚaxiss      úl/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_manipulation/split.pyr   r      s   € õ Ô˜SÐ"5°tÑ<Ô<Ð<ó    c                 óX   — | j         dk    rt          d¦  «        ‚t          | |d¦  «        S )zÅSplits an array into multiple sub arrays along the third axis.

    This is equivalent to ``split`` with ``axis=2``.

    .. seealso:: :func:`cupy.split` for more detail, :func:`numpy.dsplit`

    é   z2Cannot dsplit an array with less than 3 dimensions©ÚndimÚ
ValueErrorÚsplit©r   r   s     r	   Údsplitr      ó1   € ð „x1‚}€}ÝÐMÑNÔNÐNÝÐ)¨1Ñ-Ô-Ð-r
   c                 ó   — | j         dk    rt          d¦  «        ‚| j         dk    rt          | |d¦  «        S t          | |d¦  «        S )a  Splits an array into multiple sub arrays horizontally.

    This is equivalent to ``split`` with ``axis=0`` if ``ary`` has one
    dimension, and otherwise that with ``axis=1``.

    .. seealso:: :func:`cupy.split` for more detail, :func:`numpy.hsplit`

    r   z&Cannot hsplit a zero-dimensional arrayé   r   r   s     r	   Úhsplitr       sO   € ð „x1‚}€}ÝÐAÑBÔBÐBØ
„x1‚}€}ÝSÐ-¨qÑ1Ô1Ð1åSÐ-¨qÑ1Ô1Ð1r
   c                 óÊ   — | j         |k    rt          d¦  «        ‚| j        |         }t          j        |¦  «        r||z  dk    rt          d¦  «        ‚t          | ||¦  «        S )a§  Splits an array into multiple sub arrays along a given axis.

    Args:
        ary (cupy.ndarray): Array to split.
        indices_or_sections (int or sequence of ints): A value indicating how
            to divide the axis. If it is an integer, then is treated as the
            number of sections, and the axis is evenly divided. Otherwise,
            the integers indicate indices to split at. Note that the sequence
            on the device memory is not allowed.
        axis (int): Axis along which the array is split.

    Returns:
        A list of sub arrays. Each array is a view of the corresponding input
        array.

    .. seealso:: :func:`numpy.split`

    zAxis exceeds ndimr   zindices_or_sections must divide the size along the axes.
If you want to split the array into non-equally-sized arrays, use array_split instead.)r   Ú
IndexErrorÚshapeÚnumpyÚisscalarr   r   )r   r   r   Úsizes       r	   r   r   1   s{   € ð& „x4ÒÐÝÐ,Ñ-Ô-Ð-ØŒ9TŒ?€Då„~Ð)Ñ*Ô*ð 4ØÐ%Ñ%¨Ò*Ð*Ýð3ñ4ô 4ð 4õ sÐ/°Ñ6Ô6Ð6r
   c                 óX   — | j         dk    rt          d¦  «        ‚t          | |d¦  «        S )zÅSplits an array into multiple sub arrays along the first axis.

    This is equivalent to ``split`` with ``axis=0``.

    .. seealso:: :func:`cupy.split` for more detail, :func:`numpy.dsplit`

    r   z2Cannot vsplit an array with less than 2 dimensionsr   r   r   s     r	   Úvsplitr   Q   r   r
   )r   )r   Úcupyr   r   r   r   r   r   © r
   r	   ú<module>r!      s€   ðØ €€€à Ð Ð Ð Ð Ð ð
=ð 
=ð 
=ð 
=ð
.ð 
.ð 
.ð2ð 2ð 2ð"7ð 7ð 7ð 7ð@
.ð 
.ð 
.ð 
.ð 
.r
   