o i@sddlZddlmZddlmmZddddddddddd dd Zd d Zd dZddZddZ ej Z ddZ ddZ ddZ dddZdS)N)_corecCdSN)shaperrT/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupy/_manipulation/dims.py r cCr)N)rrrrrrr r r cCsd|Srrrrrr r cCr)N)rrrrrrrr r r cCs d|dSrrrrrr r s cCs|dSrrrrrr r r ))rr)r)r r)r)rr)rr cCs`g}|D] }t|}|j|krt||jf|j}|j|}||qt|dkr.|\}|S)z)Helper function for atleast_nd functions.r)cupyasarrayndim_atleast_nd_shape_maprreshapeappendlen)narysresa new_shaperrr _atleast_nd_helpers     rcG 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` rrrrrr atleast_1d"s rcGr)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` r rrrrr atleast_2d4s r cGr)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` rrrrrr atleast_3dHs r!cGstt|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)argsrrr broadcast_arrayses r&cCs t||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)arrayrrrr r'ts r'cCs"t|ttfvr |f}t||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_dimsraxisrrr expand_dimss r/cCs ||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` )squeezer-rrr r0s r0)N)rr!cupy._core._routines_manipulation_routines_manipulationr+rrrr r!r#r&r'r/r0rrrr s&