`i DddlZddlZddlmZdZddZddZdZdZdS) N)sparsect|tj|d}||jdk}|j||j||j|fS)aReturns the indices and values of the nonzero elements of a matrix Args: A (cupy.ndarray or cupyx.scipy.sparse.spmatrix): Matrix whose nonzero elements are desired. Returns: tuple of cupy.ndarray: It returns (``I``, ``J``, ``V``). ``I``, ``J``, and ``V`` contain respectively the row indices, column indices, and values of the nonzero matrix entries. .. seealso:: :func:`scipy.sparse.find` Tcopyr) _check_A_typer coo_matrixsum_duplicatesdatarowcol)Anz_masks o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/sparse/_extract.pyfindrsc!!$'''AfkG 5>15>16'? ::ct|tj|d}|j|z|jk}t |||S)aReturns the lower triangular portion of a matrix in sparse format Args: A (cupy.ndarray or cupyx.scipy.sparse.spmatrix): Matrix whose lower triangular portion is desired. k (integer): The top-most diagonal of the lower triangle. format (string): Sparse format of the result, e.g. 'csr', 'csc', etc. Returns: cupyx.scipy.sparse.spmatrix: Lower triangular portion of A in sparse format. .. seealso:: :func:`scipy.sparse.tril` Frrrrr r _masked_cooasformatr kformatmasks rtrilrW!!%(((A 519 D q$   ( ( 0 00rct|tj|d}|j|z|jk}t |||S)aReturns the upper triangular portion of a matrix in sparse format Args: A (cupy.ndarray or cupyx.scipy.sparse.spmatrix): Matrix whose upper triangular portion is desired. k (integer): The bottom-most diagonal of the upper triangle. format (string): Sparse format of the result, e.g. 'csr', 'csc', etc. Returns: cupyx.scipy.sparse.spmatrix: Upper triangular portion of A in sparse format. .. seealso:: :func:`scipy.sparse.triu` Frrrs rtriur2rrct|tjs5tjj|sd}t|dSdS)Nz5A must be cupy.ndarray or cupyx.scipy.sparse.spmatrix) isinstancecupyndarraycupyxscipyr isspmatrix TypeError)r msgs rrrGsU q$, ' '5;+=+H+H+K+KEnnrc|j|}|j|}|j|}tj|||ff|j|jS)N)shapedtype)r r r rrr(r))r rr r r s rrrMsH %+C %+C 6$r,s ;;;,1111*1111* OOOOOr