o Ō0 ić@srddlZddlZddlmZddd„Zddd„Ze ddd d ”Zddefd d „Zdd d„Z ddd„Z ddd„Z dS)éN)Ś_corecCs¢t|tjƒr |j}nt |”}|dkrt |”}|dkr!t |”}|dkrA|jt|ƒ}tj||f|j d}||  |”dd…<|S|dkrMt |  |””St dƒ‚)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.) Ś isinstanceŚcupyZndarrayŚndimŚnumpyŚarrayŚsizeŚabsZzerosrZdiagonalŚ ValueError)ŚvŚkr r Śret©rśa/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/cupy/_creation/matrix.pyŚdiags    rcCs$t |”r t |”}t | ”|”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 ZisscalarŚasarrayrrZravel)rrrrrŚdiagflat/s rzint32 m, int32 kzT outzK int row = i / m; int col = i % m; out = (col <= row + k); Zcupy_tricCs*|dur|}tj||f|d}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)ŚNŚMrrŚoutrrrŚtriNs rcCs:t |”}t|jdd…|tdœŽ}t |||j d””S)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` éž’’’N©rrr©rrrŚshapeŚboolŚwhererŚtype©ŚmrŚmaskrrrŚtrilfs r'cCs>t |”}t|jdd…|dtdœŽ}t ||j d”|”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` rNrrrrr$rrrŚtriu{s r(FcCsŒt |”}|jdkrtdƒ‚|durt|ƒ}tjt|ƒ|ft |jt ”d}|s3|dd…ddd…fn|}tj |  dd”t  |”|d|S)aęReturns 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) rrr rŚlenrr Z promote_typesrŚintŚpowerZreshapeZarange)ŚxrZ increasingrŚtmprrrŚvanders   r/)r)NF) r rrrrZElementwiseKernelrŚfloatrr'r(r/rrrrŚs   (ų