o Ö°“iˆã@sœddlZddlmZddlmmZdd„dd„dd„dd„dd„d d„d œZd d „Zd d„Zdd„Zdd„Z ej Z dd„Z dd„Z dd„Z ddd„ZdS)éN)Ú_corecCódS©N)é©©ÚshaperrúK/home/ubuntu/.local/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)ÚcupyÚasarrayÚndimÚ_atleast_nd_shape_maprÚreshapeÚappendÚlen)ÚnÚarysÚresÚaÚ new_shaperrr Ú_atleast_nd_helpers     rcGó td|ƒS)a­Converts 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` r©r©rrrr Ú 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|ŽjƒS)zØBroadcasts 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Ú broadcastÚvalues)Ú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` )ÚtypeÚtupler"Ú _manipulationÚ _expand_dims©rÚaxisrrr Ú expand_dims„s r/cCs | |¡S)a¸Removes 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 r0—s r0)N)rrÚ!cupy._core._routines_manipulationÚ_routines_manipulationr+rrrr r!r#r&r'r/r0rrrr Ús& ú