a 7Sic @sdZddlmZddlZddlmZmZddZddZeeeed d d Zd d Z ddZ ddZ eeeeedddZ eeedddZ ddZd"eeeeeefdddZddeeeeefdd d!ZdS)#z:Various linear algebra utility methods for internal use. )TensorN)OptionalTuplecCsDt|tjr|jtjkSd}tjs8|dt|7}t |dS)z$Check if tensor A is a sparse tensorzexpected Tensorz but got {}N) isinstancetorchrlayout sparse_coojit is_scriptingformattype TypeError)AZ error_strrO/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/_linalg_utils.py is_sparse s    rcCs$|j}|tjtjtjfvr|StjS)zTReturn the floating point dtype of tensor A. Integer types map to float32. )dtyperfloat16float32float64)rrrrrget_floating_dtypesr)rBreturncCs.|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)rrsparsemmmatmul)rrrrrr s rcCs|r|S|S)z]Return conjugate of tensor A. .. note:: If A's dtype is not complex, A is returned. ) is_complexconjrrrr conjugate-srcCst|j}||d|dS)z9Return transpose of a matrix or batches of matrices. )lenshape transpose)rndimrrrr$7s r$cCs tt|S)zCReturn transpose conjugate of a matrix or batches of matrices. )rr$rrrr transjugate>sr&)XrYrcCstt|t||S)z7Return bilinear form of matrices: :math:`X^T A Y`. )rr$)r'rr(rrrbformDsr)rScCs t|||S)z+Return quadratic form :math:`S^T A S`. )r)r*rrrqformJsr,cCs*|jrtj|j}ntjt|}|S)z*Return orthogonal basis of A columns. )is_cudarlinalgqrQorgqrgeqrf)rr0rrrbasisPsr3F)rlargestrcCsH|dur d}tjj|dd\}}|r@tj|dd}tj|dd}||fS)z4Return eigenpairs of A with specified ordering. NFU)UPLO))dims)rr.eighflip)rr4EZrrrsymeig[sr=)out)inputrrcCs tddS)NzxThis function was deprecated since version 1.9 and is now removed. Please use the `torch.linalg.solve` function instead.) RuntimeError)r?rr>rrrsolveisrA)F)__doc__rrtypingrrrrrrr$r&r)r,r3boolr=rArrrrs