o i @sRddlZejdkrddlmZn ddlmmZddlZddlm Z ddZ dS)N2) index_tricks)internalcOs|j}t||}t||d}t|jdd}dd|D}zt|}Wn t y2t dw|||g|Ri|} t | } t |jdd| j| j } | | |<|D]} ||| g|Ri|} t | | | <qZt| jD] } t| d|} qv| 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` Ncss|]}|tfVqdS)N)Ellipsis).0indr Q/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupy/lib/_shape_base.py 'sz#apply_along_axis..z;Cannot apply_along_axis when any iteration dimensions are 0)ndimr_normalize_axis_indexcupymoveaxisrndindexshapenext StopIteration ValueErrorasarrayemptydtyperange)func1daxisarrargskwargsr inarr_viewindsind0resbuffroutir r r apply_along_axis s,    r%) numpy __version__ numpy.librnumpy.lib._index_tricks_impllib_index_tricks_implr cupy._corerr%r r r r s