`i TddlZddlZddlZddlmZd dZdZdZd dZd d Z d Z dS) N)internalcVtj||j}t||S)aReverse the order of elements in an array along the given axis. Note that ``flip`` function has been introduced since NumPy v1.12. The contents of this document is the same as the original one. Args: a (~cupy.ndarray): Input array. axis (int or tuple of int or None): Axis or axes along which to flip over. The default, ``axis=None``, will flip over all of the axes of the input array. If axis is negative it counts from the last to the first axis. If axis is a tuple of ints, flipping is performed on all of the axes specified in the tuple. Returns: ~cupy.ndarray: Output array. .. seealso:: :func:`numpy.flip` )r_normalize_axis_indicesndim_flip)aaxisaxess p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_manipulation/rearrange.pyflipr s&(  +D!& 9 9D D>>cT|jdkrtd|dddddfS)aAFlip array in the left/right direction. Flip the entries in each row in the left/right direction. Columns are preserved, but appear in a different order than before. Args: a (~cupy.ndarray): Input array. Returns: ~cupy.ndarray: Output array. .. seealso:: :func:`numpy.fliplr` Input must be >= 2-dNr ValueErrorrs r fliplrr!s6 vzz/000 RRR2X;r cL|jdkrtd|dddS)a;Flip array in the up/down direction. Flip the entries in each column in the up/down direction. Rows are preserved, but appear in a different order than before. Args: a (~cupy.ndarray): Input array. Returns: ~cupy.ndarray: Output array. .. seealso:: :func:`numpy.flipud` zInput must be >= 1-dNrrrs r flipudr5s- vzz/000 TTrT7Nr cn|;t|djSt j|r|fn|}t fd|D}t|tj rk|}tt||j }t j ||f}tj ||f}g}tjD]b}dgjz}j|||<|tjj||ct%||D]0\}}||xx|zcc<||xxj|zcc<1tjD]%}tj ||j||<&t |St j||} | jdkrt+ddtjD} | D]\} }| |xx| z cc<t-dt-dffgjz} | D]^\}} | j|pdz} | rEt-d| t-| dft-| dt-d| ff| |<_tj}t3j| D]}t%|\}}|||<|S)a@Roll array elements along a given axis. Elements that roll beyond the last position are re-introduced at the first. Args: a (~cupy.ndarray): Array to be rolled. shift (int or tuple of int): The number of places by which elements are shifted. If a tuple, then `axis` must be a tuple of the same size, and each of the given axes is shifted by the corresponding number. If an int while `axis` is a tuple of ints, then the same value is used for all given axes. axis (int or tuple of int or None): The axis along which elements are shifted. By default, the array is flattened before shifting, after which the original shape is restored. Returns: ~cupy.ndarray: Output array. .. seealso:: :func:`numpy.roll` NrcDg|]}tj|jS)r_normalize_axis_indexr).0axrs r zroll..cs579&r1622r rz4'shift' and 'axis' should be scalars or 1D sequencesci|]}|dS)rr)rrs r zroll..s000B"a000r )rollravelreshapeshapenumpyisscalartuple isinstancecupyndarraymaxlensize broadcast_torangerappendarangezip broadcastndrsliceitems empty_like itertoolsproduct)rshiftr r n_axesindicesr ind_shapes broadcastedshiftsshrollsoffsetresult arr_index res_indexs` r r"r"Is\, |AGGIIua((00999nT** 4D77D =A  D%&&+ SYY ++!$ 22!%&33-- H HBaf IGBKIbM NN4;qwr{33;;IFF G G G Gu%% ' 'EB BKKK1 KKK BKKK172; &KKKK-- B BB+GBKAAGBKKw  oeT22 >A  JLL L00%--000!  FB 2JJJ" JJJJ;;d ,./!&8 ,,.. J JJB agbk&Q &F J#D6'22E&$4G4GH#VGT22E$4G4GHJb ## (%0 - -G#&= Iy !) F9   r rrrcH|j}|dkrtdt|}t|dkrtd|d|dks"t |d|dz |krtd| |dcxkr|krnn| |dcxkr|ksntd| |fz|dz}|dkr |d d S|dkrt ||St td|}||d||dc||d<||d<|dkr*tj t ||df|St tj |||dfS) a-Rotate an array by 90 degrees in the plane specified by axes. Note that ``axes`` argument has been introduced since NumPy v1.12. The contents of this document is the same as the original one. Args: a (~cupy.ndarray): Array of two or more dimensions. k (int): Number of times the array is rotated by 90 degrees. axes: (tuple of ints): The array is rotated in the plane defined by the axes. Axes must be different. Returns: ~cupy.ndarray: Output array. .. seealso:: :func:`numpy.rot90` rrzlen(axes) must be 2rrzaxes must be differentzaxes must be >= %d and < %dN) rrr(r-absrlistr0r* transpose)rkr a_ndimaxes_ts r rot90rQs$VF zz/000 ;;D 4yyA~~./// Aw$q'Sa47!233v==1222 GtAw ' ' ' ' ' ' ' ' 'VGtAw,G,G,G,G,G,G,G,G66'69JJKKK AAAvvt AvvQ~~ %6"" # #F'-d1gtAw$F47OVDG_Avv~eAQz22F;;;T^Av..a ;;;r ctdg|jz}|D]}tddd||<|t|S)Nr)r6rr()rr indexerrs r rrsMT{{maf$G,,D$++ U7^^ r )N)rrH) r9r&r* cupy._corerr rrr"rQrrr r rUs 0((HHHHV+<+<+<+<\r