`i ^ddlZejdkrddlmZn ddlmcmZddlZddlm Z dZ dS)N2) index_tricks)internalc|j}tj||}tj||d}t j|jdd}d|D} t|}n#t$rtdwxYw|||g|Ri|} tj | } tj |jdd| jz| j } | | |<|D]+} ||| g|Ri|} tj | | | <,t| jD]} tj| d|} | S)aApply a function to 1-D slices along the given axis. Args: func1d (function (M,) -> (Nj...)): This function should accept 1-D arrays. It is applied to 1-D slices of ``arr`` along the specified axis. It must return a 1-D ``cupy.ndarray``. axis (integer): Axis along which ``arr`` is sliced. arr (cupy.ndarray (Ni..., M, Nk...)): Input array. args: Additional arguments for ``func1d``. kwargs: Additional keyword arguments for ``func1d``. Returns: cupy.ndarray: The output array. The shape of ``out`` is identical to the shape of ``arr``, except along the ``axis`` dimension. This axis is removed, and replaced with new dimensions equal to the shape of the return value of ``func1d``. So if ``func1d`` returns a scalar ``out`` will have one fewer dimensions than ``arr``. .. seealso:: :func:`numpy.apply_along_axis` Nc3,K|]}|tfzVdS)N)Ellipsis).0inds h/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/lib/_shape_base.py z#apply_along_axis..'s) . .#C8+  . . . . . .z;Cannot apply_along_axis when any iteration dimensions are 0)ndimr_normalize_axis_indexcupymoveaxisrndindexshapenext StopIteration ValueErrorasarrayemptydtyperange)func1daxisarrargskwargsr inarr_viewindsind0resbuffr outis r apply_along_axisr( s* 8D  )$ 5 5DsD"--J   0" 5 6 6D . . . . .D Dzz     I     &D! 3D 3 3 3F 3 3C ,s  C :j&ss+ci7 C CDDJ&&fZ_6t666v66L%%S 38__--}T2t,, Ks !A11B ) numpy __version__ numpy.librnumpy.lib._index_tricks_impllib_index_tricks_implr cupy._corerr(rr r1s s&&&&&&&777777777 88888r