`i/ddlZddlmZmZddlZddlZddlmZmZddl m Z m Z m Z m Z mZedfde de de d eefd Z d d d d Z d!deede de dddee d eef dZddedfdedeedede de d eef dZefdede d eefdZedfde de de d eefdZ d!deede de dddee d eef dZedfde de de d eefdZ d!deede de dddee d eef dZ d"de dede de d eef dZ d!deedede de dddee d eefdZdS)#N)AnyOptional)_get_strides_for_order_K_update_order_char) _OrderKACF_OrderCF _ShapeLike DTypeLikeNDArrayCshapedtypeorderreturnc0tj|||S)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 rrs h/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_creation/basic.pyemptyr s& <uE 2 2 22T) get_memptrc H|}|dvr"td|tj|r|f}|dkr|t ||jkrdStt|j j |j j t|}|dkrft|tj||}d}|tj|n|j}|rt%j||jnd}|||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_contiguousordrrmathprodsizerrdata)arrr rstridesr,memptrs r_new_like_order_and_stridesr1#s2 KKMME (((3::5AABBB ~e ||)c%jjAF.B.B " ag2CJJ@@ A AE ||*1ek%.@.@%HH#(#4ty!&7AKD...33tgv%%dD  rr prototypesubokc|td||j}t||||\}}}|r|n|j}t j|||||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) TypeErrorrr1r rr)r2rrr3r r/r0s r empty_liker7Csl> 4555 }85%((E7F /EE E <ufgu = ==rNMkc||}t||f||}|| 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)rr)zerosdiagonalfill)r8r9r:rrrets reyerAmsa4 y  Ae5 1 1 1CQBww!q&& LLOO Jrnc$t||S)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)rA)rBrs ridentityrDs q   rc^tj|||}|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` rr<)rrr?r rrr.s ronesrGs.* UE///AFF1III Hrr.c|td||j}t||||\}}}|r|n|j}t j|||||}|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` Nr5r<)r6rr1r rrr?r.rrr3r r/r0s r ones_likerJs< 4555 }8E59>@@E7F 'EEE UE67E::AFF1III Hrcttj|||}|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_asyncnbytesrFs rr=r=s9& UE///AF18$$$ Hrc|td||j}t||||\}}}|r|n|j}t j|||||}|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` Nr5r) r6rr1r rrr-rLrMrIs r zeros_likerOs< 4555 }8E59>@@E7F 'EEE UE67E::AF18$$$ Hr fill_valuec|;t|tjr|j}nt j|j}tj|||}tj||d|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) isinstancerrrr!arraycopyto)r rPrrr.s rfullrX*sk. } j$, / / 2$EEK ++1E UE///AK:x0000 Hrc|td||j}t||||\}}}|r|n|j}t j|||||}t j||d|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` Nr5rRrS)r6rr1r rrrW)r.rPrrr3r r/r0s r full_likerZKs@ 4555 }8E59>@@E7F 'EEE UE67E::AK:x0000 Hr)N)NrNN)Nr )r*typingrrr!rcupy._core.internalrrcupy.typing._typesrrr r r floatrr1r7intrArDrGrJr=rOrXrZrrrasR LLLLLLLL!33333 S\ 3333. $!37!!!!!D &* '>'>3<'>'>'> '>  # '>  S\ '>'>'>'>X       C=        S\    F).I'#,(!      S\     8 &* ( ( 3<( ( (  (  # (  S\ ( ( ( ( Z!      S\     4 &* ( ( 3<( ( (  (  # (  S\ ( ( ( ( \          S\     H &* * * 3<* * *  *  *  # *  S\* * * * * * r