`inbddlZddlmZdZd ddddZdZddddZdddd Zd dddd ZdS) N)_corec>td|Drtdt|}t|D]F\}}|jdkr|ddt jf}|||<,|jdkrtdGt|dS)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` c3LK|]}t|tj V dS)N) isinstancecupyndarray.0as k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_manipulation/join.py zcolumn_stack..s1 8 8qz!T\** * 8 8 8 8 8 8z&Only cupy arrays can be column stackedNz4Only 1 or 2 dimensional arrays can be column stacked)axis) any TypeErrorlist enumeratendimrnewaxis ValueError concatenate)tuplstir s r column_stackrs 8 8C 8 8 888B@AAA s))C#HH1 6Q;;!!!T\/"ACFF Vq[[FHH H s # # ##r same_kinddtypecastingcR|d|D}d}tj|||||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` Nc6g|]}|S)ravelr ms r zconcatenate..:s &&&Qqwwyy&&&rr)rconcatenate_methodrroutr r!s r rr#s:, |&&#&&&  #CsE7 C CCrc6td|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` c6g|]}tj|Sr$)r atleast_3dr&s r r(zdstack..L"888q**888rrr)rs r dstackr1?s# 88C888! < <.ds" , , ,1DOA   , , ,rrrr)rr)rr r!arrsrs r hstackr6OsI* - , , , ,D D Aw|q tT @ @ @@rc<td|Dd||S)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` c6g|]}tj|Sr$)r atleast_2dr&s r r(zvstack..r/rrrr0)rr r!s r vstackr:ks3, 88C888!"G 5 5 55rcDtfd|D|||S)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` c:g|]}tj|Sr$)r expand_dims)r xrs r r(zstack..s&???a(D11???rrr0r*s ` r stackr?s<$ ????3???s"G 5 5 55r)rN)rrrrr1r6r:r?r$rr r@s $$$<DkDDDDD8 = = = {AAAAA8{5555545$ 5555555r