o 0 i/@s4ddlZddlmZmZddlZddlZddlmZmZddl m Z m Z m Z m Z mZedfde de de d eefd d Z d+d d ddZ    d,deede de dddee d eef ddZddedfdedeedede de d eef ddZefdede d eefddZedfde de de d eefddZ    d,deede de dddee d eef d d!Zedfde de de d eefd"d#Z    d,deede de dddee d eef d$d%Z  d-de d&ede de d eef d'd(Z    d,deed&ede de dddee d eefd)d*ZdS).N)AnyOptional)_get_strides_for_order_K_update_order_char) _OrderKACF_OrderCF _ShapeLike DTypeLikeNDArrayCshapedtypeorderreturncCstj|||dS)aReturns an array without initializing the elements. Args: shape (int or tuple of ints): Dimensionalities of the array. dtype (data-type, optional): Data type specifier. order ({'C', 'F'}): Row-major (C-style) or column-major (Fortran-style) order. Returns: cupy.ndarray: A new array with elements not initialized. .. seealso:: :func:`numpy.empty` r)cupyndarray)r r rr`/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupy/_creation/basic.pyempty srT) get_memptrcCs|}|dvrtd|t|r|f}|dkr(|dur(t||jkr(dStt|j j |j j t |}|dkrbt |t||}d}|durNt|n|j}|r[tj||djnd}|||fS|ddfS)zw Determine order and strides as in NumPy's PyArray_NewLikeArray. (see: numpy/core/src/multiarray/ctors.c) )r FKAzorder not understood: {}rN)r NNr r )upper ValueErrorformatnumpyZisscalarlenndimchrrflags c_contiguous f_contiguousordrr mathprodsizerrdata)ar rr rstridesr(memptrrrr_new_like_order_and_strides#s"   r-r prototypesubokcCsR|durtd|dur|j}t||||\}}}|r|n|j}t|||||S)aReturns a new array with same shape and dtype of a given array. This function currently does not support ``subok`` option. Args: a (cupy.ndarray): Base array. dtype (data-type, optional): Data type specifier. The data type of ``a`` is used by default. order ({'C', 'F', 'A', or 'K'}): Overrides the memory layout of the result. ``'C'`` means C-order, ``'F'`` means F-order, ``'A'`` means ``'F'`` if ``a`` is Fortran contiguous, ``'C'`` otherwise. ``'K'`` means match the layout of ``a`` as closely as possible. subok: Not supported yet, must be None. shape (int or tuple of ints): Overrides the shape of the result. If ``order='K'`` and the number of dimensions is unchanged, will try to keep order, otherwise, ``order='C'`` is implied. Returns: cupy.ndarray: A new array with same shape and dtype of ``a`` with elements not initialized. .. seealso:: :func:`numpy.empty_like` Nsubok is not supported yet) TypeErrorr r-r rr)r.r rr/r r+r,rrr empty_likeCs r2NMkcCsH|dur|}t||f||d}|| ks||kr|S||d|S)aReturns a 2-D array with ones on the diagonals and zeros elsewhere. Args: N (int): Number of rows. M (int): Number of columns. ``M == N`` by default. k (int): Index of the diagonal. Zero indicates the main diagonal, a positive index an upper diagonal, and a negative index a lower diagonal. dtype (data-type, optional): Data type specifier. order ({'C', 'F'}): Row-major (C-style) or column-major (Fortran-style) order. Returns: cupy.ndarray: A 2-D array with given diagonals filled with ones and zeros elsewhere. .. seealso:: :func:`numpy.eye` N)r r)zerosZdiagonalfill)r3r4r5r rretrrreyemsr:ncCs t||dS)a'Returns a 2-D identity array. It is equivalent to ``eye(n, n, dtype)``. Args: n (int): Number of rows and columns. dtype (data-type, optional): Data type specifier. Returns: cupy.ndarray: A 2-D identity array. .. seealso:: :func:`numpy.identity` r)r:)r;r rrridentitys r<cCstj|||d}|d|S)aReturns a new array of given shape and dtype, filled with ones. This function currently does not support ``order`` option. Args: shape (int or tuple of ints): Dimensionalities of the array. dtype (data-type, optional): Data type specifier. order ({'C', 'F'}): Row-major (C-style) or column-major (Fortran-style) order. Returns: cupy.ndarray: An array filled with ones. .. seealso:: :func:`numpy.ones` rr6)rrr8r r rr*rrroness r>r*cCs`|durtd|dur|j}t||||\}}}|r|n|j}t|||||}|d|S)aReturns an array of ones with same shape and dtype as a given array. This function currently does not support ``subok`` option. Args: a (cupy.ndarray): Base array. dtype (data-type, optional): Data type specifier. The dtype of ``a`` is used by default. order ({'C', 'F', 'A', or 'K'}): Overrides the memory layout of the result. ``'C'`` means C-order, ``'F'`` means F-order, ``'A'`` means ``'F'`` if ``a`` is Fortran contiguous, ``'C'`` otherwise. ``'K'`` means match the layout of ``a`` as closely as possible. subok: Not supported yet, must be None. shape (int or tuple of ints): Overrides the shape of the result. If ``order='K'`` and the number of dimensions is unchanged, will try to keep order, otherwise, ``order='C'`` is implied. Returns: cupy.ndarray: An array filled with ones. .. seealso:: :func:`numpy.ones_like` Nr0r6)r1r r-r rrr8r*r rr/r r+r,rrr ones_likes  r@cCs$tj|||d}|jd|j|S)aReturns a new array of given shape and dtype, filled with zeros. Args: shape (int or tuple of ints): Dimensionalities of the array. dtype (data-type, optional): Data type specifier. order ({'C', 'F'}): Row-major (C-style) or column-major (Fortran-style) order. Returns: cupy.ndarray: An array filled with zeros. .. seealso:: :func:`numpy.zeros` rr)rrr) memset_asyncnbytesr=rrrr7sr7cCsf|durtd|dur|j}t||||\}}}|r|n|j}t|||||}|jd|j|S)aReturns an array of zeros with same shape and dtype as a given array. This function currently does not support ``subok`` option. Args: a (cupy.ndarray): Base array. dtype (data-type, optional): Data type specifier. The dtype of ``a`` is used by default. order ({'C', 'F', 'A', or 'K'}): Overrides the memory layout of the result. ``'C'`` means C-order, ``'F'`` means F-order, ``'A'`` means ``'F'`` if ``a`` is Fortran contiguous, ``'C'`` otherwise. ``'K'`` means match the layout of ``a`` as closely as possible. subok: Not supported yet, must be None. shape (int or tuple of ints): Overrides the shape of the result. If ``order='K'`` and the number of dimensions is unchanged, will try to keep order, otherwise, ``order='C'`` is implied. Returns: cupy.ndarray: An array filled with zeros. .. seealso:: :func:`numpy.zeros_like` Nr0r) r1r r-r rrr)rArBr?rrr zeros_likes rC fill_valuecCsL|durt|tjr|j}nt|j}tj|||d}tj||dd|S)a!Returns a new array of given shape and dtype, filled with a given value. This function currently does not support ``order`` option. Args: shape (int or tuple of ints): Dimensionalities of the array. fill_value: A scalar value to fill a new array. dtype (data-type, optional): Data type specifier. order ({'C', 'F'}): Row-major (C-style) or column-major (Fortran-style) order. Returns: cupy.ndarray: An array filled with ``fill_value``. .. seealso:: :func:`numpy.full` NrunsafeZcasting) isinstancerrr rarraycopyto)r rDr rr*rrrfull*s  rJcCsf|durtd|dur|j}t||||\}}}|r|n|j}t|||||}tj||dd|S)aReturns a full array with same shape and dtype as a given array. This function currently does not support ``subok`` option. Args: a (cupy.ndarray): Base array. fill_value: A scalar value to fill a new array. dtype (data-type, optional): Data type specifier. The dtype of ``a`` is used by default. order ({'C', 'F', 'A', or 'K'}): Overrides the memory layout of the result. ``'C'`` means C-order, ``'F'`` means F-order, ``'A'`` means ``'F'`` if ``a`` is Fortran contiguous, ``'C'`` otherwise. ``'K'`` means match the layout of ``a`` as closely as possible. subok: Not supported yet, must be None. shape (int or tuple of ints): Overrides the shape of the result. If ``order='K'`` and the number of dimensions is unchanged, will try to keep order, otherwise, ``order='C'`` is implied. Returns: cupy.ndarray: An array filled with ``fill_value``. .. seealso:: :func:`numpy.full_like` Nr0rErF)r1r r-r rrrI)r*rDr rr/r r+r,rrr full_likeKs  rK)N)NrNN)Nr )r&typingrrrrZcupy._core.internalrrZcupy.typing._typesrrrr r floatrr-r2intr:r<r>r@r7rCrJrKrrrrs   " , #  -  . $