`iT. ddlZddlZddlZddlmZddZddZejddddZd Zej d d d d dddZ ddZ ej dZ ddddZ ddZdS)N)_corec |y|}t|tjr|jtjkr ||Stj|jt}d||<||St|tjr-|jtjkrtj |||Stj|j |t}d||<tj |||S)a Delete values from an array along the specified axis. Args: arr (cupy.ndarray): Values are deleted from a copy of this array. indices (slice, int or array of ints): These indices correspond to values that will be deleted from the copy of `arr`. Boolean indices are treated as a mask of elements to remove. axis (int or None): The axis along which `indices` correspond to values that will be deleted. If `axis` is not given, `arr` will be flattened. Returns: cupy.ndarray: A copy of `arr` with values specified by `indices` deleted along `axis`. .. warning:: This function may synchronize the device. .. seealso:: :func:`numpy.delete`. NdtypeFaxis) ravel isinstancecupyndarrayrbool_onessizeboolcompressshape)arrindicesrmasks q/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_manipulation/add_remove.pydeleters2 |iikk gt| , , !$*1L1Lx= y...W 4y gt| , , ;$*1L1L='3T::: :y4555W }T3T2222ctj|}tj|}|Mtj||fdStj||f|S)a Append values to the end of an array. Args: arr (array_like): Values are appended to a copy of this array. values (array_like): These values are appended to a copy of ``arr``. It must be of the correct shape (the same shape as ``arr``, excluding ``axis``). If ``axis`` is not specified, ``values`` can be any shape and will be flattened before use. axis (int or None): The axis along which ``values`` are appended. If ``axis`` is not given, both ``arr`` and ``values`` are flattened before use. Returns: cupy.ndarray: A copy of ``arr`` with ``values`` appended to ``axis``. Note that ``append`` does not occur in-place: a new array is allocated and filled. If ``axis`` is None, ``out`` is a flattened array. .. seealso:: :func:`numpy.append` Nr)r asarrayrconcatenate_methodr )rvaluesrs rappendr9sv2 ,s  C \& ! !F |' YY[[&,,.. )1...3egg 6  #S&M4 8 88rzraw T x, int64 sizezT yzy = x[i % size] cupy_resizec,tj|rtj||Stj|}|jdkrtj||jStj||j}t||j||S)aReturn a new array with the specified shape. If the new array is larger than the original array, then the new array is filled with repeated copies of ``a``. Note that this behavior is different from a.resize(new_shape) which fills with zeros instead of repeated copies of ``a``. Args: a (array_like): Array to be resized. new_shape (int or tuple of int): Shape of resized array. Returns: cupy.ndarray: The new array is formed from the data in the old array, repeated if necessary to fill out the required number of elements. The data are repeated in the order that they are stored in memory. .. seealso:: :func:`numpy.resize` rr) numpyisscalarr fullrrzerosrempty_resize_kernel)a new_shapeouts rresizer)as( ~a'yA&&& QAv{{z)173333 *Y ( (C1afc""" JrzT data, int64 lenzint64 yzdata == T(0) ? len : _jz min(a, b)zy = alen first_nonzerofbc~|jdkrtd|jstdd}|j}|}d|vr't ||j}d|vr8|jt |ddd|jz }|||S) aTrim the leading and/or trailing zeros from a 1-D array or sequence. Returns the trimmed array Args: filt(cupy.ndarray): Input array trim(str, optional): 'fb' default option trims the array from both sides. 'f' option trim zeros from front. 'b' option trim zeros from back. Returns: cupy.ndarray: trimmed input .. seealso:: :func:`numpy.trim_zeros` z'Multi-dimensional trim is not supportedz0-d array cannot be trimmedrFBN)ndim ValueError TypeErrorrupper_first_nonzero_krnlitem)filttrimstartends r trim_zerosr<s$ y1}}BCCC 975666 E )C ::<.compare_axis_elemssoTlF4Lete|S11 :a=1  7D dj.. t!8OurTr1rF) _unique_1dr moveaxisrarangeintpreshapemathprodascontiguousarray issubdtyperunsignedinteger iscomplexobjastyper$tolistpopranger*rrr anyrsumr7cumsumnonzero)arrGrHrIrrEret orig_shapeidx is_unsignedrTsorted_indicesqueuecurrentoffmid_elemrNrOielem_posrimaskinv_idxrgrRrSs @@runiquervsp |,(6'4#,... r4 # #BJ +aAdi 8 8 8C JqM49Z^#<#< = =B  # #B/"(D,@AAK"2&&J F&49%%Z 1 TY???NjjllA  E 2++yy||  b== 1:q#g,,'' ) )A!!'!*h77 ) GAJ'''' WQZ((((T? dC[!!! eX\*+++#+x % 2++( N B w{{z"($*555RaRabb6RW$QRRx1%%%y"(1+dj999QRR DB DHHJJOO%% 7 122 7 7 7B r1d # #B #C% ~d#$$ D!!A%*TZty999"' wx#,t$$Q'j',*,gm<<CRC)B s122wSbS!"" 3xx1}}!f Jrc8tj|}|s|r|}||}n||}tj|jtj}d|dd<|dd|ddk|dd<|r t|dd|dd||}|s|s|s|S|f}|r |||fz }|rGtj |dz } tj|jtj } | | |<|| fz }|rgtj |d} tj| j dzf| j } | | dd<|j | d<|| dd| ddz fz }|S)NrTr.r1r)r rflattenargsortsortr$rr rDrfrXrgrr) rhrGrHrIrEpermauxrrirtrurgrks rrUrUGs b   ! ! # #B~zz||h   :citz 2 2 2DD!H122w#crc("DH:%d122hCRC999 d)C } $C tDz{ D!!A%*TZty999  wx#,t$$Q'j',*,gm<<CRC)B s122wSbS!"" Jrr>)r,)FFFN)FFFT)r r rZrrrElementwiseKernelr%r)ReductionKernelr6r<fusionfuserDrvrUrrrsM +3+3+3+3b9999B)(5<,e+   @,,, 38%)T8<TTTTTn7<.2""""""r