9j)SSKJr SSKrSSKrSSKJr SSjrSSjr\R"SSSS 5rSS\ 4S jr SS jr SS jr SS jr g)) annotationsN)_corec[U[R5(a URnON[R"U5nUS:Xa[R "U5nUS:Xa[R "U5nUS:XaMUR [U5-n[R"X34URS9nXRU5SS&U$US:Xa%[R "URU55$[S5e)a\Returns 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.) isinstancecupyndarrayndimnumpyarraysizeabszerosr diagonal ValueError)vkr rrets O/home/wildlama/miniconda3/lib/python3.13/site-packages/cupy/_creation/matrix.pydiagr s&!T\\""vvzz!} 19 1 A 19 AA qyvvAjj$QWW5 Q zz!**Q-((344c[R"U5(a[R"U5n[R"UR 5U5$)a)Creates 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` )risscalarasarrayr rravel)rrs rdiagflatr1s8 ~~a MM!  99QWWY ""rzint32 m, int32 kzT outzK int row = i / m; int col = i % m; out = (col <= row + k); cupy_tricPUcUn[R"X4US9n[XU5$)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` r)r empty _tri_kernel)NMrr outs rtrir'Ps-" y  **aV5 )C qS !!rc[R"U5n[URSSU[S.6n[R "X UR RS55$)aYReturns 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` Nrr rr rr'shapeboolwherer typemrmasks rtrilr3hsF QA   .D ::dqww||A //rc[R"U5n[URSSUS- [S.6n[R "X R RS5U5$)a[Returns 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+r0s rtriur5}sJ QA  !4 0D ::dGGLLOQ //rc[R"U5nURS:wa [S5eUc [ U5n[R "[ U5U4[ R"UR[5S9nU(d USS2SSS24OUn[R"URSS5[R"U5US9 U$)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_typesr intpowerreshapearange)xr$ increasingrtmps rvanderrAs" QAvv{IJJy F CFA;e&9&9!''3&GHA&!AttG*ACJJqyyQQS9 Hr)r)NF) __future__rrr rrrElementwiseKernelr#floatr'r3r5rArrrFs[" %5P#&%%    Qe"00*0* r