9jSSKJr SSKrSSKrSSKJr SrS SSS.SjjrSrSSS.S jrSSS.S jr SSS.S jr S SSS.S jjr g)) annotationsN)_corec([SU55(a [S5e[U5n[U5HMup#URS:XaUSS2[ R 4nX1U'M2URS:wdMD[S5e [USS9$)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` c3`# UH$n[U[R5(+v M& g7f)N) isinstancecupyndarray).0as Q/home/wildlama/miniconda3/lib/python3.13/site-packages/cupy/_manipulation/join.py column_stack..s! 8Cqz!T\\* * *Cs,.z&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_stackr s 8C 888@AA s)C# 66Q;!T\\/"AF VVq[FH H  s ## same_kinddtypecastingcUc!UVs/sHoUR5PM nnSn[R"XX#U5$s snf)uJoins 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` r)ravelrconcatenate_method)rroutr!r"ms r rr's@, |"%&#Qwwy#&  # #Cs7 CC's=cn[UVs/sHn[R"U5PM snS5$s snf)zStacks 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` r)rr atleast_3d)rr's r dstackr*Cs, C8Cq*C8! <<8s 2cUVs/sHn[R"U5PM nnSnUSRS:XaSn[XEXS9$s snf)uhStacks 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` rrr )r atleast_1drr)rr!r"r arrsrs r hstackr.SsM*), ,1DOOA D , D Aw||q t @@ -s Acl[UVs/sHn[R"U5PM snSXS9$s snf)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` rr )rr atleast_2d)rr!r"r's r vstackr1os3, C8Cq*C8!" 558 1cJ[R"S[SS9 [XUS9$)Nz:`row_stack` alias is deprecated. Use `cp.vstack` directly.r) stacklevelr )warningswarnDeprecationWarningr1)rr!r"s r row_stackr8s( MM $  #G 44rc l[UVs/sHn[R"XQ5PM snXX4S9$s snf)uCStacks 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` r )rr expand_dims)rrr&r!r"xs r stackr<s5$ 3?3a((13?" 55?r2)rN) __future__rr5rrrrr*r.r1r8r<rr r?s^" $<DkD8 = {A8{54!+55$ 5r