o װi @sLddlZddlZddlmZddZd ddZd ddZd d Zd d ZdS)N)sparsecCsDt|tj|dd}||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_maskr O/home/ubuntu/.local/lib/python3.10/site-packages/cupyx/scipy/sparse/_extract.pyfinds  rcCs6t|tj|dd}|j||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 kformatmaskr r rtrilrcCs6t|tj|dd}|j||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` Frrrr r rtriu2rrcCs.t|tjstjj|sd}t|dSdS)Nz5A must be cupy.ndarray or cupyx.scipy.sparse.spmatrix) isinstancecupyndarraycupyxscipyr isspmatrix TypeError)r msgr r rrGsrcCs:|j|}|j|}|j|}tj|||ff|j|jdS)N)shapedtype)r r rrrr"r#)r rr r rr r rrMs   r)rN) rr cupyx.scipyrrrrrrr r r rs