a d@sVdZddlmZmZddlZddlmZddZddZeeeed d d Zd d Z ddZ ddZ eeeeedddZ eeedddZ ddZd.eeeeeefdddZd/ddedd d!Zddeeeeefd"d#d$Zddeeeeefd"d%d&Zd0ddeeefdd(d)Zd1ddd*eeeeefd+d,d-ZdS)2z:Various linear algebra utility methods for internal use. )OptionalTupleN)TensorcCsDt|tjr|jtjkSd}tjs8|dt|7}t |dS)z$Check if tensor A is a sparse tensorzexpected Tensorz but got {}N) isinstancetorchrZlayoutZ sparse_cooZjitZ is_scriptingformattype TypeError)AZ error_strr \/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/_linalg_utils.py is_sparse s    r cCs$|j}|tjtjtjfvr|StjS)zTReturn the floating point dtype of tensor A. Integer types map to float32. )dtyperfloat16float32float64)r rr r r get_floating_dtypesr)r BreturncCs.|dur |St|r"tj||St||S)ziMultiply two matrices. If A is None, return B. A can be sparse or dense. B is always dense. N)r rsparsemmmatmul)r rr r r r!s rcCs|r|S|S)z]Return conjugate of tensor A. .. note:: If A's dtype is not complex, A is returned. )Z is_complexZconjr r r r conjugate.srcCst|j}||d|dS)z4Return transpose of a matrix or batches of matrices.)lenshape transpose)r ndimr r r r8s rcCs tt|S)z>Return transpose conjugate of a matrix or batches of matrices.)rrrr r r transjugate>sr )Xr YrcCstt|t||S)z2Return bilinear form of matrices: :math:`X^T A Y`.)rr)r!r r"r r r bformCsr#r ScCs t|||S)z&Return quadratic form :math:`S^T A S`.)r#r$r r r qformHsr&cCstj|jS)z%Return orthogonal basis of A columns.)rlinalgZqrQrr r r basisMsr)F)r largestrcCsH|dur d}tjj|dd\}}|r@tj|dd}tj|dd}||fS)z/Return eigenpairs of A with specified ordering.NFU)ZUPLO))dims)rr'ZeighZflip)r r*EZr r r symeigRsr0)out)rcCstdddS)NBThis function was deprecated since version 1.9 and is now removed.z;Please use the `torch.linalg.matrix_rank` function instead. RuntimeError)inputZtolZ symmetricr1r r r matrix_rank`sr6)r5r rcCs tddS)NzxThis function was deprecated since version 1.9 and is now removed. Please use the `torch.linalg.solve` function instead.r3r5r r1r r r solvegsr8cCstdddS)Nr2z5Please use the `torch.linalg.lstsq` function instead.r3r7r r r lstsqmsr9TcCs tddS)NzwThis function was deprecated since version 1.9 and is now removed. Please use the `torch.linalg.eigh` function instead.r3)r5 eigenvectorsupperr1r r r _symeigtsr<)ev)selfr:rcCs tddS)NzvThis function was deprecated since version 1.9 and is now removed. Please use the `torch.linalg.eig` function instead.r3)r?r:r=r>r r r eig|sr@)F)NF)FT)F)__doc__typingrrrrr rrrrr r#r&r)boolr0r6r8r9r<r@r r r r s: