o 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) isinstancecupyndarray.0ar T/home/ubuntu/veenaModal/venv/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 enumeratendimrnewaxis ValueError concatenate)tuplstirr r r column_stacks    r 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 )ravelrmr r r :szconcatenate..r)rconcatenate_methodrroutrrr r r 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 )r atleast_3dr!r r r r#Lzdstack..r r)rr r r 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 )r atleast_1drr r r r#dr)zhstack..r rr)rr)rrrarrsrr r r 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 )r atleast_2dr!r r r r#r)zvstack..rrr*)rrrr r r vstackksr0cs 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 )r expand_dims)rxrr r r#szstack..rr*r%r rr stacksr3)rN)rrrrr+r.r0r3r r r r s