`i~ddlZddlZddlmZddZddZejddddZddefd Zdd Z dd Z dd Z dS)N)_corect|tjr|j}nHt j|}|dkrtj|}|dkrt j|}|dkrP|jt|z}tj||f|j }|| |dd<|S|dkr'tj| |Std)aReturns a diagonal or a diagonal array. Args: v (array-like): Array or array-like object. k (int): Index of diagonals. Zero indicates the main diagonal, a positive value an upper diagonal, and a negative value a lower diagonal. Returns: cupy.ndarray: If ``v`` indicates a 1-D array, then it returns a 2-D array with the specified diagonal filled by ``v``. If ``v`` indicates a 2-D array, then it returns the specified diagonal of ``v``. In latter case, if ``v`` is a :class:`cupy.ndarray` object, then its view is returned. .. seealso:: :func:`numpy.diag` dtypeNzInput must be 1- or 2-d.) isinstancecupyndarrayndimnumpyarraysizeabszerosrdiagonal ValueError)vkr rrets i/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/_creation/matrix.pydiagrs&!T\""vz!}} 199 1 A 199 AA qyyvAj$QW555 Q z!**Q--(((3444ctj|rtj|}tj||S)aECreates a diagonal array from the flattened input. Args: v (array-like): Array or array-like object. k (int): Index of diagonals. See :func:`cupy.diag` for detail. Returns: cupy.ndarray: A 2-D diagonal array with the diagonal copied from ``v``. .. seealso:: :func:`numpy.diagflat` )r isscalarasarrayr rravel)rrs rdiagflatr/s? ~a M!   9QWWYY " ""rzint32 m, int32 kzT outzK int row = i / m; int col = i % m; out = (col <= row + k); cupy_tric\||}tj||f|}t|||S)aCreates an array with ones at and below the given diagonal. Args: N (int): Number of rows. M (int): Number of columns. ``M == N`` by default. k (int): The sub-diagonal at and below which the array is filled. Zero is the main diagonal, a positive value is above it, and a negative value is below. dtype: Data type specifier. Returns: cupy.ndarray: An array with ones at and below the given diagonal. .. seealso:: :func:`numpy.tri` Nr)r empty _tri_kernel)NMrrouts rtrir&Ns8" y  *aV5 ) ) )C q!S ! !!rctj|}t|jdd|td}tj|||jdS)a}Returns a lower triangle of an array. Args: m (array-like): Array or array-like object. k (int): The diagonal above which to zero elements. Zero is the main diagonal, a positive value is above it, and a negative value is below. Returns: cupy.ndarray: A lower triangle of an array. .. seealso:: :func:`numpy.tril` Nrrrr rr&shapeboolwherertypemrmasks rtrilr2fsO QA   . . .D :dAqw||A / //rctj|}t|jdd|dz td}tj||jd|S)aReturns an upper triangle of an array. Args: m (array-like): Array or array-like object. k (int): The diagonal below which to zero elements. Zero is the main diagonal, a positive value is above it, and a negative value is below. Returns: cupy.ndarray: An upper triangle of an array. .. seealso:: :func:`numpy.triu` r(Nrr)rr*r/s rtriur4{sS QA  !4 0 0 0D :dAGLLOOQ / //rFctj|}|jdkrtd|t |}tjt ||ft j|jt}|s|dddddfn|}tj | ddtj |||S)aReturns a Vandermonde matrix. Args: x (array-like): 1-D array or array-like object. N (int, optional): Number of columns in the output. ``N = len(x)`` by default. increasing (bool, optional): Order of the powers of the columns. If True, the powers increase from right to left, if False (the default) they are reversed. Returns: cupy.ndarray: A Vandermonde matrix. .. seealso:: :func:`numpy.vander` rz.x must be a one-dimensional array or sequence.Nr)r%) r rr rlenr!r promote_typesrintpowerreshapearange)xr# increasingrtmps rvanderr@s" QAv{{IJJJy FF CFFA;e&9!'3&G&GHHHA& -!AAAtttG**ACJqyyQQS9999 Hr)r)NF) r r rrrElementwiseKernelr"floatr&r2r4r@rrrDs %5%5%5%5P####&&e%     Qe""""00000*0000*      r