o 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 rrQ/home/ubuntu/veenaModal/venv/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 ValueErrorformatnumpyisscalarlenndimchrrflags 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 rr0r r,r-rrr empty_likeCs r3NMkcCsH|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)zerosdiagonalfill)r4r5r6r 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` rr7)rrr:r 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` Nr1r7)r2r r.r rrr:r+r rr0r r,r-rrr ones_likes  rBcCs$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?rrrr8sr8cCsf|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` Nr1r) r2r r.r rrr*rCrDrArrr zeros_likes rE 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` Nrunsafecasting) isinstancerrr rarraycopyto)r rFr rr+rrrfull*s  rMcCsf|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` Nr1rGrH)r2r r.r rrrL)r+rFr rr0r r,r-rrr full_likeKs  rN)N)NrNN)Nr )r'typingrrrrcupy._core.internalrrcupy.typing._typesrrrr r floatrr.r3intr<r>r@rBr8rErMrNrrrrs   " , #  -  . $