`iddlZddlmZddlmcmZddddddd Zd Zd Zd Zd Z ej Z dZ dZ dZ ddZdS)N)_corecdSN)shapes k/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_manipulation/dims.pyr s$cdS)N)rrrrs r r r s&r c d|zSrrrs r r r s $,r cdS)N)rrrrrs r r r s)r cd|zdzSrrrs r r r s$,-r c |dzSrrrs r r r s %$,r ))rr)r)rr)r)rr)rrcg}|D]a}tj|}|j|kr+t||jf|j}|j|}||bt|dkr|\}|S)z)Helper function for atleast_nd functions.r)cupyasarrayndim_atleast_nd_shape_mapr reshapeappendlen)narysresa new_shapes r _atleast_nd_helperr!s C  LOO 6A::-q!&k:17CCI 9%A 1  3xx1}} Jr c"td|S)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"td|S)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'4s" a & &&r c"td|S)a?Converts 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)Hs. a & &&r c8tt|jS)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` )list broadcastvalues)argss r broadcast_arraysr/es  4 ' ( ((r c,tj||S)aBroadcast 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 r1r1ts  eU + ++r clt|ttfvr|f}tj||S)a3Expands 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` )typetupler+ _manipulation _expand_dimsraxiss r expand_dimsr:s3 Dzz%&&u  %a . ..r c,||S)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` )squeezer8s r r<r<s 99T??r )N)rr!cupy._core._routines_manipulation_routines_manipulationr6rr!r%r'r)r,r/r1r:r<rr r r?s 999999999   & & # # - - & &     '''$'''('''4 O  ) ) ) , , , ///&r