o Ô0 i ã@s^ddlZddlZddlZddlmZddd„Zdd„Zdd„Zdd d „Zdd d„Z dd„Z dS)éN)ÚinternalcCst ||j¡}t||ƒS)aÉReverse 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` )rZ_normalize_axis_indicesÚndimÚ_flip)ÚaÚaxisÚaxes©rúh/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupy/_manipulation/rearrange.pyÚflip s r cCs(|jdkr tdƒ‚|dd…ddd…fS)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-dNéÿÿÿÿ©rÚ ValueError©rrrr Úfliplr!s rcCs |jdkr tdƒ‚|ddd…S)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-dNr rrrrr Úflipud5s rcs<|durtˆ ¡|dƒ ˆj¡St |¡r|fn|}t‡fdd„|Dƒƒ}t|tj ƒr | ¡}t t |ƒ|j ƒ}t  ||f¡}t  ||f¡}g}tˆjƒD]}dgˆj}ˆj|||<| t ˆj|¡ |¡¡qLt||ƒD]\}}|||8<||ˆj|;<qotˆjƒD] }t  ||ˆj¡||<qŒˆt|ƒSt ||¡} | jdkr¯tdƒ‚dd„tˆjƒDƒ} | D] \} }| || 7<q»tdƒtdƒffgˆj} |  ¡D](\}} | ˆj|päd;} | rtd| ƒt| dƒft| dƒtd| ƒff| |<qÙt ˆ¡}tj| ŽD]}t|Ž\}}ˆ|||<q |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` Nrcsg|] }t |ˆj¡‘qSr)rZ_normalize_axis_indexr©Ú.0Úaxrrr Ú csÿzroll..rz4'shift' and 'axis' should be scalars or 1D sequencescSsi|]}|d“qS)rrrrrr Ú €szroll..)ÚrollZravelZreshapeÚshapeÚnumpyZisscalarÚtupleÚ isinstanceÚcupyZndarrayÚmaxÚlenÚsizeZ broadcast_toÚrangerÚappendZarangeÚzipÚ broadcastÚndrÚsliceÚitemsZ empty_likeÚ itertoolsÚproduct)rÚshiftrrZn_axesÚindicesrZ ind_shapeÚsZ broadcastedZshiftsÚshZrollsÚoffsetÚresultZ arr_indexZ res_indexrrr rIsV ÿ     ÿ ÿ€  rr©rrcCsL|j}|dkr tdƒ‚t|ƒ}t|ƒdkrtdƒ‚|d|dks-t|d|dƒ|kr1tdƒ‚| |dkr>|krNnn| |dkrM|ksWntd| |fƒ‚|d}|dkre|d d …S|dkrnt||ƒSttd|ƒƒ}||d||d||d<||d<|dkršt  t||dfƒ|¡Stt  ||¡|dfƒS) 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` r r zlen(axes) must be 2rrzaxes must be differentzaxes must be >= %d and < %déN) rrrr ÚabsrÚlistr"rZ transpose)rÚkrZa_ndimZaxes_trrr Úrot90”s( (:  *r6cCs6tdƒg|j}|D] }tdddƒ||<q |t|ƒS)Nr )r'rr)rrZindexerrrrr rÂs r)N)rr1) r)rrZ cupy._corerr rrrr6rrrrr Ús    K .