9jbSSKJr SSKrSSKJr S SjrSrS SjrS SjrS Sjr S r SS jr g)) annotationsN)internalc&URXU5$)ajTakes elements of an array at specified indices along an axis. This is an implementation of "fancy indexing" at single axis. This function does not support ``mode`` option. Args: a (cupy.ndarray): Array to extract elements. indices (int or array-like): Indices of elements that this function takes. axis (int): The axis along which to select indices. The flattened input is used by default. out (cupy.ndarray): Output array. If provided, it should be of appropriate shape and dtype. Returns: cupy.ndarray: The result of fancy indexing. .. seealso:: :func:`numpy.take` )take)aindicesaxisouts Q/home/wildlama/miniconda3/lib/python3.13/site-packages/cupy/_indexing/indexing.pyrrs. 66' %%cURRS;a [S5eUcUR5nSnURn[ R "X#5nX1R:wa [S5e/n[UR5HdupVXR:XaURU5 MSU-S-SX5- S- --nUR[R"U5RU55 Mf U[U5$)aWTake values from the input array by matching 1d index and data slices. Args: a (cupy.ndarray): Array to extract elements. indices (cupy.ndarray): Indices to take along each 1d slice of ``a``. axis (int): The axis to take 1d slices along. Returns: cupy.ndarray: The indexed result. .. seealso:: :func:`numpy.take_along_axis` )iuz"`indices` must be an integer arrayrz9`indices` and `a` must have the same number of dimensions))r)dtypekind IndexErrorravelndimr_normalize_axis_index ValueError enumerateshapeappendcupyarangereshapetuple)rrr r fancy_indexrn ind_shapes r take_along_axisr#!s}}+=>> | GGI 66D  ) )$ 5D || GI IK!''" 9   w 'q5(448a<+@@I   t{{1~55i@ A # U;   r c&URXU5$)N)choose)rchoicesr modes r r%r%Is 88G$ ''r c&URXU5$)aReturns selected slices of an array along given axis. Args: condition (1-D array of bools): Array that selects which entries to return. If len(condition) is less than the size of a along the given axis, then output is truncated to the length of the condition array. a (cupy.ndarray): Array from which to extract a part. axis (int): Axis along which to take slices. If None (default), work on the flattened array. out (cupy.ndarray): Output array. If provided, it should be of appropriate shape and dtype. Returns: cupy.ndarray: A copy of a without the slices along axis for which condition is false. .. warning:: This function may synchronize the device. .. seealso:: :func:`numpy.compress` )compress) conditionrr r s r r)r)Ms4 ::is ++r c&URXU5$)aReturns specified diagonals. This function extracts the diagonals along two specified axes. The other axes are not changed. This function returns a writable view of this array as NumPy 1.10 will do. Args: a (cupy.ndarray): Array from which the diagonals are taken. offset (int): Index of the diagonals. Zero indicates the main diagonals, a positive value upper diagonals, and a negative value lower diagonals. axis1 (int): The first axis to take diagonals from. axis2 (int): The second axis to take diagonals from. Returns: cupy.ndarray: A view of the diagonals of ``a``. .. seealso:: :func:`numpy.diagonal` )diagonal)roffsetaxis1axis2s r r,r,js, ::fU ++r cD[U[R5(d [S5e[U[R5(d[R"U5nUR 5nUR 5nUR UR5S5$)aSReturn the elements of an array that satisfy some condition. This is equivalent to ``np.compress(ravel(condition), ravel(arr))``. If ``condition`` is boolean, ``np.extract`` is equivalent to ``arr[condition]``. Args: condition (int or array_like): An array whose nonzero or True entries indicate the elements of array to extract. a (cupy.ndarray): Input array of the same size as condition. Returns: cupy.ndarray: Rank 1 array of values from arr where condition is True. .. warning:: This function may synchronize the device. .. seealso:: :func:`numpy.extract` z/extract requires input array to be cupy.ndarrayr) isinstancerndarray TypeErrorarrayrrnonzero)r*rs r extractr6st, a & &IJJ i . .JJy)   A!I 66)##%a( ))r c[U5[U5:wa [S5e[U5S:Xa [S5e[R"U5(d [ S5e[ [U55Htn[ X[R5(d [ S5eXnURR[RLdM\[SRU55e [R"U6n[R"U6n[R"/UQUP76nUSRS:XaUSRnO*[R"USUS5SRn[R "Xe5n[R""Xr5 USSS 2nUSSS 2n[%X5Hup[R""XxUS 9 M U$) aZReturn an array drawn from elements in choicelist, depending on conditions. Args: condlist (list of bool arrays): The list of conditions which determine from which array in `choicelist` the output elements are taken. When multiple conditions are satisfied, the first one encountered in `condlist` is used. choicelist (list of cupy.ndarray): The list of arrays from which the output elements are taken. It has to be of the same length as `condlist`. default (scalar) : If provided, will fill element inserted in `output` when all conditions evaluate to False. default value is 0. Returns: cupy.ndarray: The output at position m is the m-th element of the array in `choicelist` where the m-th element of the corresponding array in `condlist` is True. .. seealso:: :func:`numpy.select` z7list of cases must be same length as list of conditionsrz3select with an empty condition list is not possiblez"default only accepts scalar valuesz.choicelist only accepts lists of cupy ndarraysz7invalid entry {} in condlist: should be boolean ndarrayNr)where)lenrrisscalarr3ranger1r2rtypebool_format result_typebroadcast_arraysrremptycopytozip) condlist choicelistdefaultrcondr result_shaperesultchoices r selectrLs, 8}J' EG G 8}NOO == ! !<== 3z? #*-66LM M{ ::??$** ,I  $   j )E$$h/H&&< rQs3" &4%!P(,:,2*D<r