9jSSKJr SSKrSSKJr SSKJs Jr SSSSSS S .rS rS r S r Sr \Rr Sr SrSrSSjrg)) annotationsN)_corecgN)shapes Q/home/wildlama/miniconda3/lib/python3.13/site-packages/cupy/_manipulation/dims.pyr s$cg)N)rrrr s r r r s&r c SU-$rrr s r r r s$,r cg)N)rrrrr s r r r s)r cSU-S-$rrr s r r r s $,-r c US-$rrr s r r r s%$,r ))rr)r)rr)r)rr)rrc/nUHmn[R"U5nURU:a3[XR4"UR5nUR "U6nUR U5 Mo [U5S:XaUunU$)z)Helper function for atleast_nd functions.r)cupyasarrayndim_atleast_nd_shape_mapr reshapeappendlen)narysresa new_shapes r _atleast_nd_helperr"sv C  LLO 66A:-q&&k:177CI 9%A 1  3x1} Jr c[SU5$)aConverts arrays to arrays with dimensions >= 1. Args: arys (tuple of arrays): Arrays to be converted. All arguments must be :class:`cupy.ndarray` objects. Only zero-dimensional array is affected. Returns: If there are only one input, then it returns its converted version. Otherwise, it returns a list of converted arrays. .. seealso:: :func:`numpy.atleast_1d` rr"rs r atleast_1dr&$s a &&r c[SU5$)aConverts arrays to arrays with dimensions >= 2. If an input array has dimensions less than two, then this function inserts new axes at the head of dimensions to make it have two dimensions. Args: arys (tuple of arrays): Arrays to be converted. All arguments must be :class:`cupy.ndarray` objects. Returns: If there are only one input, then it returns its converted version. Otherwise, it returns a list of converted arrays. .. seealso:: :func:`numpy.atleast_2d` rr$r%s r atleast_2dr(6s" a &&r c[SU5$)aConverts arrays to arrays with dimensions >= 3. If an input array has dimensions less than three, then this function inserts new axes to make it have three dimensions. The place of the new axes are following: - If its shape is ``()``, then the shape of output is ``(1, 1, 1)``. - If its shape is ``(N,)``, then the shape of output is ``(1, N, 1)``. - If its shape is ``(M, N)``, then the shape of output is ``(M, N, 1)``. - Otherwise, the output is the input array itself. Args: arys (tuple of arrays): Arrays to be converted. All arguments must be :class:`cupy.ndarray` objects. Returns: If there are only one input, then it returns its converted version. Otherwise, it returns a list of converted arrays. .. seealso:: :func:`numpy.atleast_3d` rr$r%s r atleast_3dr*Js. a &&r c8[[U6R5$)zBroadcasts given arrays. Args: args (tuple of arrays): Arrays to broadcast for each other. Returns: list: A list of broadcasted arrays. .. seealso:: :func:`numpy.broadcast_arrays` )tuple broadcastvalues)argss r broadcast_arraysr0gs D!(( ))r c.[R"X5$)zBroadcast an array to a given shape. Args: array (cupy.ndarray): Array to broadcast. shape (tuple of int): The shape of the desired array. Returns: cupy.ndarray: Broadcasted view. .. seealso:: :func:`numpy.broadcast_to` )r broadcast_to)arrayr s r r2r2vs   e ++r cf[U5[[4;aU4n[R"X5$)aExpands given arrays. Args: a (cupy.ndarray): Array to be expanded. axis (int): Position where new axis is to be inserted. Returns: cupy.ndarray: The number of dimensions is one greater than that of the input array. .. seealso:: :func:`numpy.expand_dims` )typer,list _manipulation _expand_dimsr axiss r expand_dimsr;s, Dz%&u  % %a ..r c$URU5$)aRemoves size-one axes from the shape of an array. Args: a (cupy.ndarray): Array to be reshaped. axis (int or tuple of ints): Axes to be removed. This function removes all size-one axes by default. If one of the specified axes is not of size one, an exception is raised. Returns: cupy.ndarray: An array without (specified) size-one axes. .. seealso:: :func:`numpy.squeeze` )squeezer9s r r=r=s 99T?r )N) __future__rrr!cupy._core._routines_manipulation_routines_manipulationr7rr"r&r(r*r-r0r2r;r=rr r rAsc" 99  & # - &   '$'('4 OO  * , /&r