o 0 in@slddlZddlmZddZddddddZd d Zdddd d Zdddd dZddddddZdS)N)_corecCsxtdd|Dr tdt|}t|D] \}}|jdkr,|ddtjf}|||<q|jdkr5tdqt|ddS) a{Stacks 1-D and 2-D arrays as columns into a 2-D array. A 1-D array is first converted to a 2-D column array. Then, the 2-D arrays are concatenated along the second axis. Args: tup (sequence of arrays): 1-D or 2-D arrays to be stacked. Returns: cupy.ndarray: A new 2-D array of stacked columns. .. seealso:: :func:`numpy.column_stack` css|] }t|tj VqdS)N) isinstancecupyZndarray.0arc/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupy/_manipulation/join.py szcolumn_stack..z&Only cupy arrays can be column stackedNz4Only 1 or 2 dimensional arrays can be column stackedaxis) any TypeErrorlist enumeratendimrZnewaxis ValueError concatenate)tuplstirrrr column_stacks    rZ same_kinddtypecastingcCs,|dur dd|D}d}t|||||S)u$Joins arrays along an axis. Args: tup (sequence of arrays): Arrays to be joined. All of these should have same dimensionalities except the specified axis. axis (int or None): The axis to join arrays along. If axis is None, arrays are flattened before use. Default is 0. out (cupy.ndarray): Output array. dtype (str or dtype): If provided, the destination array will have this dtype. Cannot be provided together with ``out``. casting ({‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}, optional): Controls what kind of data casting may occur. Defaults to ``'same_kind'``. Returns: cupy.ndarray: Joined array. .. seealso:: :func:`numpy.concatenate` NcSsg|]}|qSr)Zravelrmrrr :szconcatenate..r)rZconcatenate_methodrroutrrrrr r#srcCstdd|DdS)aStacks arrays along the third axis. Args: tup (sequence of arrays): Arrays to be stacked. Each array is converted by :func:`cupy.atleast_3d` before stacking. Returns: cupy.ndarray: Stacked array. .. seealso:: :func:`numpy.dstack` cSg|]}t|qSr)rZ atleast_3drrrr rLzdstack..r r)rrrr dstack?s r%cCs4dd|D}d}|djdkrd}t||||dS)uStacks arrays horizontally. If an input array has one dimension, then the array is treated as a horizontal vector and stacked along the first axis. Otherwise, the array is stacked along the second axis. Args: tup (sequence of arrays): Arrays to be stacked. dtype (str or dtype): If provided, the destination array will have this dtype. casting ({‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}, optional): Controls what kind of data casting may occur. Defaults to ``'same_kind'``. Returns: cupy.ndarray: Stacked array. .. seealso:: :func:`numpy.hstack` cSr"r)rZ atleast_1drrrr rdr#zhstack..r rr)rr)rrrZarrsrrrr hstackOs r&cCstdd|Dd||dS)uStacks arrays vertically. If an input array has one dimension, then the array is treated as a horizontal vector and stacked along the additional axis at the head. Otherwise, the array is stacked along the first axis. Args: tup (sequence of arrays): Arrays to be stacked. Each array is converted by :func:`cupy.atleast_2d` before stacking. dtype (str or dtype): If provided, the destination array will have this dtype. casting ({‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}, optional): Controls what kind of data casting may occur. Defaults to ``'same_kind'``. Returns: cupy.ndarray: Stacked array. .. seealso:: :func:`numpy.dstack` cSr"r)rZ atleast_2drrrr rr#zvstack..rrr$)rrrrrr vstackksr'cs tfdd|D|||dS)uwStacks arrays along a new axis. Args: tup (sequence of arrays): Arrays to be stacked. axis (int): Axis along which the arrays are stacked. out (cupy.ndarray): Output array. dtype (str or dtype): If provided, the destination array will have this dtype. Cannot be provided together with ``out``. casting ({‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}, optional): Controls what kind of data casting may occur. Defaults to ``'same_kind'``. Returns: cupy.ndarray: Stacked array. .. seealso:: :func:`numpy.stack` csg|]}t|qSr)rZ expand_dims)rxr rr rszstack..rr$r rr r stacksr))rN)rrrrr%r&r'r)rrrr s