U Kc @s8dZddlmZmZddlZddlmZddZddZeeeed d d Zd d Z ddZ ddZ eeeeedddZ eeedddZ ddZd+eeeeeefdddZd,ddedd d!Zddeeeeefd"d#d$Zddeeeeefd"d%d&Zd-ddd'eeeeefd(d)d*ZdS).z: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 7/tmp/pip-unpacked-wheel-gikjz4vx/torch/_linalg_utils.py is_sparse s    r cCs$|j}|tjtjtjfkr|StjS)zTReturn the floating point dtype of tensor A. Integer types map to float32. )dtyperZfloat16Zfloat32Zfloat64)r rr r r get_floating_dtypesr)r BreturncCs.|dkr |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)rr rr r r bformCsr r ScCs t|||S)z&Return quadratic form :math:`S^T A S`.)r r!r r r qformHsr#cCs*|jrtj|j}ntjt|}|S)z%Return orthogonal basis of A columns.)Zis_cudarlinalgZqrQZorgqrZgeqrf)r r%r r r basisMsr&F)r largestrcCsH|dkr d}tjj|dd\}}|r@tj|dd}tj|dd}||fS)z/Return eigenpairs of A with specified ordering.NFU)ZUPLO))Zdims)rr$ZeighZflip)r r'EZr r r symeigWsr,)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 symmetricr-r r r matrix_rankesr2)r1r rcCs tddS)NzxThis function was deprecated since version 1.9 and is now removed. Please use the `torch.linalg.solve` function instead.r/r1r r-r r r solvelsr4cCstdddS)Nr.z5Please use the `torch.linalg.lstsq` function instead.r/r3r r r lstsqrsr5)ev)self eigenvectorsrcCs tddS)NzvThis function was deprecated since version 1.9 and is now removed. Please use the `torch.linalg.eig` function instead.r/)r8r9r6r7r r r eigysr:)F)NF)F)__doc__typingrrrrr rrrrrr r#r&boolr,r2r4r5r:r r r r s2