`i%ddlZddlZddlmZddlmZddlmZddlm Z ej dddd Z dd Z dd Zdd ZddZddZddZddZdZddZddZdS)N)_routines_logic)_fusion_thread_local)search)_utilzraw T X, int64 lenzbool zzBz = (i == 0 || X[i] != X[i-1]) && (i == len - 1 || X[i] != X[i+1]) setxorkernelFctjr3|rtdtjtj|||St j|d||||S)a#Tests whether all array elements along a given axis evaluate to True. Parameters ---------- a : cupy.ndarray Input array. axis : int or tuple of ints Along which axis to compute all. The flattened array is used by default. out : cupy.ndarray Output array. keepdims : bool If ``True``, the axis is remained as an axis of size one. Returns ------- y : cupy.ndarray An array reduced of the input array along the axis. See Also -------- numpy.all z3cupy.all does not support `keepdims` in fusion yet.axisoutaarg_namer r keepdims)r is_fusingNotImplementedErrorcall_reduction_logicallr check_arrayr r r rs e/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_logic/truth.pyrr2%''/  G%EGG G#2 J#/// / a#&&&& 55dh5 7 77ctjr3|rtdtjtj|||St j|d||||S)a#Tests whether any array elements along a given axis evaluate to True. Parameters ---------- a : cupy.ndarray Input array. axis : int or tuple of ints Along which axis to compute all. The flattened array is used by default. out : cupy.ndarray Output array. keepdims : bool If ``True``, the axis is remained as an axis of size one. Returns ------- y : cupy.ndarray An array reduced of the input array along the axis. See Also -------- numpy.any z3cupy.any does not support `keepdims` in fusion yet.r r r r)rrrrranyrrrs rrr7rrcv|}|}|jdks |jdkrL|r%tj|jtjStj|jtjStj|}tj |||j|S)aTests whether each element of a 1-D array is also present in a second array. Returns a boolean array the same length as ``ar1`` that is ``True`` where an element of ``ar1`` is in ``ar2`` and ``False`` otherwise. Parameters ---------- ar1 : cupy.ndarray Input array. ar2 : cupy.ndarray The values against which to test each value of ``ar1``. assume_unique : bool, optional Ignored invert : bool, optional If ``True``, the values in the returned array are inverted (that is, ``False`` where an element of ``ar1`` is in ``ar2`` and ``True`` otherwise). Default is ``False``. Returns ------- y : cupy.ndarray, bool The values ``ar1[in1d]`` are in ``ar2``. rdtype) ravelsizecupyonesshapebool_zerossort_search_exists_kernel)ar1ar2 assume_uniqueinverts rin1dr.\s6 ))++C ))++C x1}}A  ;9SYdj999 9:citz::: : )C..C  !#sCHf = ==rc|s^|r3tj|d\}}tj|d\}}nQtj|}tj|}n(|}|}|s$tj|||jd}||Stj|||jd\}}||}tj|} ||} |s|| } || } || | fS)aoFind the intersection of two arrays. Returns the sorted, unique values that are in both of the input arrays. Parameters ---------- arr1, arr2 : cupy.ndarray Input arrays. Arrays will be flattened if they are not in 1D. assume_unique : bool By default, False. If set True, the input arrays will be assumed to be unique, which speeds up the calculation. If set True, but the arrays are not unique, incorrect results and out-of-bounds indices could result. return_indices : bool By default, False. If True, the indices which correspond to the intersection of the two arrays are returned. Returns ------- intersect1d : cupy.ndarray Sorted 1D array of common and unique elements. comm1 : cupy.ndarray The indices of the first occurrences of the common values in `arr1`. Only provided if `return_indices` is True. comm2 : cupy.ndarray The indices of the first occurrences of the common values in `arr2`. Only provided if `return_indices` is True. See Also -------- numpy.intersect1d T) return_indexF)r"uniquer r(r)r!_exists_and_searchsorted_kernel flatnonzero) arr1arr2r,return_indicesind1ind2maskv1int1d arr1_indices arr2_indicess r intersect1dr>sB    %T===JD$T===JD$$;t$$D;t$$DDzz||zz|| %dD$)UCCDz6 dDIu&&HD" JE#D))Ld8L *L) L) , ,,rcXt|||||jS)aCalculates element in ``test_elements``, broadcasting over ``element`` only. Returns a boolean array of the same shape as ``element`` that is ``True`` where an element of ``element`` is in ``test_elements`` and ``False`` otherwise. Parameters ---------- element : cupy.ndarray Input array. test_elements : cupy.ndarray The values against which to test each value of ``element``. This argument is flattened if it is an array or array_like. assume_unique : bool, optional Ignored invert : bool, optional If ``True``, the values in the returned array are inverted, as if calculating element not in ``test_elements``. Default is ``False``. Returns ------- y : cupy.ndarray, bool Has the same shape as ``element``. The values ``element[isin]`` are in ``test_elements``. r,r-)r.reshaper$)element test_elementsr,r-s risinrDs38 m   &ww}556rc|rtj|}n(tj|}tj|}|t||ddS)aFind the set difference of two arrays. It returns unique values in `ar1` that are not in `ar2`. Parameters ---------- ar1 : cupy.ndarray Input array ar2 : cupy.ndarray Input array for comparison assume_unique : bool By default, False, i.e. input arrays are not unique. If True, input arrays are assumed to be unique. This can speed up the calculation. Returns ------- setdiff1d : cupy.ndarray Returns a 1D array of values in `ar1` that are not in `ar2`. It always returns a sorted output for unsorted input only if `assume_unique=False`. See Also -------- numpy.setdiff1d Tr@)r"r r1r.)r*r+r,s r setdiff1drFsU6jook#k# tCD>>> ??rc F|s(tj|}tj|}tj||fd}|jdkr|S||t ||jtj|jtjS)a1Find the set exclusive-or of two arrays. Parameters ---------- ar1, ar2 : cupy.ndarray Input arrays. They are flattened if they are not already 1-D. assume_unique : bool By default, False, i.e. input arrays are not unique. If True, input arrays are assumed to be unique. This can speed up the calculation. Returns ------- setxor1d : cupy.ndarray Return the sorted, unique values that are in only one (not both) of the input arrays. See Also -------- numpy.setxor1d Nr rr)r"r1 concatenater!r' _setxorkernelr&r%)r*r+r,auxs rsetxor1drLs. k#k#  C:D 1 1 1C x1}} HHJJJ }S#(!Z CCCEE FFrcVtjtj||fdS)aFind the union of two arrays. Returns the unique, sorted array of values that are in either of the two input arrays. Parameters ---------- arr1, arr2 : cupy.ndarray Input arrays. They are flattened if they are not already 1-D. Returns ------- union1d : cupy.ndarray Sorted union of the input arrays. See Also -------- numpy.union1d NrH)r"r1rI)r4r5s runion1drN)s(* ;t't 4@@@ A AArcZtjdtt||||S)NzPlease use `all` instead.)warningswarnDeprecationWarningrrs ralltruerSA* M-/ABBB q$X & &&rcZtjdtt||||S)NzPlease use `any` instead.)rPrQrRrrs rsometruerVFrTr)NNF)FF)F)rPr" cupy._corerrr cupy._sortingrr(r_coreElementwiseKernelrJrrr.r>rDrFrLrNrSrVrrr\sc 000000++++++++++++ ,, H  "8"8"8"8J"8"8"8"8J%>%>%>%>P:-:-:-:-z6666@ @ @ @ @F"F"F"F"FJBBB0'''' ''''''r