o 0cDB@sddlmZddlmZmZddlmZddlmZer*ddl m Z m Z m Z m Z mZddlmZddlZddlZGd d d eZGd d d eZGd ddeZGdddeZdddiddZdddjd d!Zdkd"d#Zdd$dld&d'Zdmd(d)Zdkd*d+Zdkd,d-Zdnd.d/Zdd0d1dod5d6Zdpd8d9Z dd:dqd=d>Z!dkd?d@Z"dndAdBZ#dd:dqdCdDZ$dEdFdrdIdJZ%dsdKdLZ&dMdNZ'dndOdPZ(dQdRdtdTdUZ)dudWdXZ*dYdZdvd]d^Z+dd$dld_d`Z,dddjdadbZ-dddYdcdwdfdgZ.gdhZ/dS)x) annotations)_floating_dtypes_numeric_dtypes)Array) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc@eZdZUded<ded<dS) EighResultr eigenvalues eigenvectorsN__name__ __module__ __qualname____annotations__rrL/var/www/html/gps/gps/lib/python3.10/site-packages/numpy/array_api/linalg.pyr  rc@r)QRResultrQRNrrrrrrrrc@r) SlogdetResultrsign logabsdetNrrrrrrrrc@s&eZdZUded<ded<ded<dS) SVDResultrUSVhNrrrrrr s  r F)upperxrr$boolreturncCs:|jtvr tdtj|j}|rt|j St|S)z Array API compatible wrapper for :py:func:`np.linalg.cholesky `. See its docstring for more information. z2Only floating-point dtypes are allowed in cholesky) dtyper TypeErrornplinalgcholesky_arrayr_newmT)r%r$Lrrrr,"s   r,axisx1x2r3intcCsr|jtvs |jtvrtd|j|jkrtd|jdkr!td|j|dkr,tdttj |j |j |dS)zz Array API compatible wrapper for :py:func:`np.cross `. See its docstring for more information. z(Only numeric dtypes are allowed in crossz"x1 and x2 must have the same shaperz/cross() requires arrays of dimension at least 1zcross() dimension must equal 3r2) r(rr)shape ValueErrorndimrr.r*crossr-)r4r5r3rrrr;2s  r;cC&|jtvr tdttj|jS)z Array API compatible wrapper for :py:func:`np.linalg.det `. See its docstring for more information. z-Only floating-point dtypes are allowed in det) r(rr)rr.r*r+detr-r%rrrr=Ds r=)offsetr?cCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r1r?axis1axis2)rr.r*diagonalr-r%r?rrrrDQsrDcC,|jtvr tdtttjtj |j S)z Array API compatible wrapper for :py:func:`np.linalg.eigh `. See its docstring for more information. z.Only floating-point dtypes are allowed in eigh) r(rr)rmaprr.r*r+eighr-r>rrrrH\ rHcCr<)z Array API compatible wrapper for :py:func:`np.linalg.eigvalsh `. See its docstring for more information. z2Only floating-point dtypes are allowed in eigvalsh) r(rr)rr.r*r+eigvalshr-r>rrrrJl rJcCr<)z Array API compatible wrapper for :py:func:`np.linalg.inv `. See its docstring for more information. z-Only floating-point dtypes are allowed in inv) r(rr)rr.r*r+invr-r>rrrrLyrKrLcCs2|jtvs |jtvrtdtt|j|jS)z| Array API compatible wrapper for :py:func:`np.matmul `. See its docstring for more information. z)Only numeric dtypes are allowed in matmul)r(rr)rr.r*matmulr-r4r5rrrrMsrMfro)keepdimsordrPrQ2Optional[Union[int, float, Literal['fro', 'nuc']]]cCs.|jtvr tdttjj|jd||dS) Array API compatible wrapper for :py:func:`np.linalg.norm `. See its docstring for more information. z5Only floating-point dtypes are allowed in matrix_norm)r@r1r3rPrQ) r(rr)rr.r*r+normr-)r%rPrQrrr matrix_norms rVncCs(|jtvr tdttj|j|S)z Array API compatible wrapper for :py:func:`np.matrix_power `. See its docstring for more information. zMOnly floating-point dtypes are allowed for the first argument of matrix_power) r(rr)rr.r*r+ matrix_powerr-)r%rWrrrrXs rX)rtolrYOptional[Union[float, Array]]cCs|jdkr tjdtjj|jdd}|dur0|jdddt|jd dt|j j }nt |t r8|j}|jdddt |d tjf}t tj||kdd S) z Array API compatible wrapper for :py:func:`np.matrix_rank `. See its docstring for more information. zA1-dimensional array given. Array must be at least two-dimensionalF compute_uvNr1T)r3rPr@.r2)r:r*r+ LinAlgErrorsvdr-maxr8finfor(eps isinstancerasarraynewaxisr. count_nonzero)r%rYr"tolrrr matrix_ranks  0 "rhcCs(|jdkr tdtt|jddS)Nr[z5x must be at least 2-dimensional for matrix_transposer1r@)r:r9rr.r*swapaxesr-r>rrrmatrix_transposes rjcCsN|jtvs |jtvrtd|jdks|jdkrtdtt|j |j S)zz Array API compatible wrapper for :py:func:`np.outer `. See its docstring for more information. z(Only numeric dtypes are allowed in outerrz/The input arrays to outer must be 1-dimensional) r(rr)r:r9rr.r*outerr-rNrrrrks rkcCsR|jtvr td|durt|jddt|jj}t tj j |j |dS)z Array API compatible wrapper for :py:func:`np.linalg.pinv `. See its docstring for more information. z.Only floating-point dtypes are allowed in pinvNr@)rcond) r(rr)r`r8r*rarbrr.r+pinvr-)r%rYrrrrms  rmreducedmoderpLiteral['reduced', 'complete']cC0|jtvr tdtttjtjj |j |dS)z Array API compatible wrapper for :py:func:`np.linalg.qr `. See its docstring for more information. z,Only floating-point dtypes are allowed in qrro) r(rr)rrGrr.r*r+qrr-)r%rprrrrs rscCrF)z Array API compatible wrapper for :py:func:`np.linalg.slogdet `. See its docstring for more information. z1Only floating-point dtypes are allowed in slogdet) r(rr)rrGrr.r*r+slogdetr-r>rrrru rIrucCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkr<| j }n| j }|| rEdnd}||}|||||d}| |j | dd S) Nr[) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->d) signatureextobjF)copy) linalg.linalgrvrwrxryrzr{r|r+r}r:solve1solveastype)abrvrwrxryrzr{r|r}_wraptresult_tgufuncr~rrrrr_solve!s$    rcCs0|jtvs |jtvrtdtt|j|jS)z Array API compatible wrapper for :py:func:`np.linalg.solve `. See its docstring for more information. z/Only floating-point dtypes are allowed in solve)r(rr)rr.rr-rNrrrr<srT full_matricesrcCrr)z Array API compatible wrapper for :py:func:`np.linalg.svd `. See its docstring for more information. z-Only floating-point dtypes are allowed in svdr) r(rr)r rGrr.r*r+r_r-)r%rrrrr_Irtr_Union[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFr\) r(rr)rr.r*r+r_r-r>rrrsvdvalsZs rr[axesr/Union[int, Tuple[Sequence[int], Sequence[int]]]cCs6|jtvs |jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr)r(rr)rr.r* tensordotr-)r4r5rrrrrbsrc Cs2|jtvr tdtttj|j|dddS)zz Array API compatible wrapper for :py:func:`np.trace `. See its docstring for more information. z(Only numeric dtypes are allowed in tracer@r1rA) r(rr)rr.r*rdtracer-rErrrrks  rc Cs|jtvs |jtvrtdt|j|j}d||jt|j}d||jt|j}||||kr9tdt |j |j \}}t ||d}t ||d}|ddddf|d}t |dS)Nz)Only numeric dtypes are allowed in vecdot)rz6x1 and x2 must have the same size along the given axisr1.).N).rr)r(rr)r`r:tupler8r9r*broadcast_arraysr-moveaxisrr.) r4r5r3r:x1_shapex2_shapex1_x2_resrrrvecdotxsrrT%Optional[Union[int, Tuple[int, ...]]]Optional[Union[int, float]]cs|jtvr td|jdurdn7ttrNtfddtjD}|}t | t fddDgfdd|DRdt t jj||dS) rSz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr.0ir2rr szvector_norm..csg|]}j|qSr)r8r)rrr szvector_norm..rT)r(rr)r-flattenrcrranger:r* transposereshapeprodrr.r+rU)r%r3rPrQrestnewshaper)rr3r vector_norms  <r)r,r;r=rDrHrJrLrMrVrXrhrjrkrmrsrurr_rrrrr)r%rr$r&r'r)r4rr5rr3r6r'r)r%rr'r)r%rr?r6r'r)r%rr'r)r4rr5rr'r)r%rrPr&rQrRr'r)r%rrWr6r'r)r%rrYrZr'r)r%rrprqr'r)r%rr'r)r%rrr&r'r )r%rr'r)r4rr5rrrr'r) r%rr3rrPr&rQrr'r)0 __future__r_dtypesrr _array_objectrtypingr_typingrr r r r r numpy.linalgnumpyr*rrrr r,r;r=rDrHrJrLrMrVrXrhrjrkrmrsrurrr_rrrrr__all__rrrrsL