a p=icDB@sddlmZddlmZmZddlmZddlmZerTddl m Z m Z m Z m Z mZddlmZddlZddlZGd d d eZGd d d eZGd ddeZGdddeZddddddddZdddddddddZddddd Zdd!dddd"d#d$Zdd dd%d&Zdddd'd(Zdddd)d*Zdddd+d,d-Zdd.d/ddd0dd1d2d3Zdddd4d5d6Z dd7dd8dd9d:d;Z!ddddd?Z#dd7dd8dd9d@dAZ$dBdCddDd dEdFdGZ%ddddHdIZ&dJdKZ'dddd+dLdMZ(dNdOddddPdQdRZ)ddSddTdUZ*dVdWdddXddYdZd[Z+dd!dddd"d\d]Z,dddddddd^d_Z-dddVd`ddaddbddcdddeZ.gdfZ/dS)g) annotations)_floating_dtypes_numeric_dtypes)Array) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc@seZdZUded<ded<dS) EighResultrZ eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrg/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/numpy/array_api/linalg.pyrs rc@seZdZUded<ded<dS)QRResultrQRNrrrrrrs rc@seZdZUded<ded<dS) SlogdetResultrsignZ logabsdetNrrrrrrs rc@s&eZdZUded<ded<ded<dS) SVDResultrUSZVhNrrrrrrs rF)upperrbool)xrreturncCs:|jtvrtdtj|j}|r0t|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_newZmT)r rLrrrr&"s   r&axisint)x1x2r,r!cCsr|jtvs|jtvrtd|j|jkr0td|jdkrBtd|j|dkrXtdttj |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 3r+) r"rr#shape ValueErrorndimrr(r$crossr')r.r/r,rrrr42s  r4)r r!cCs&|jtvrtdttj|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 rrrr5Ds r5)offset)r r7r!cCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r*r7Zaxis1Zaxis2)rr(r$diagonalr'r r7rrrr:Qsr:cCs,|jtvrtdtttjtj |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'r6rrrr=\s r=cCs&|jtvrtdttj|jS)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'r6rrrr>ls r>cCs&|jtvrtdttj|jS)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'r6rrrr?ys r?)r.r/r!cCs2|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'r.r/rrrr@sr@Zfro)keepdimsordz2Optional[Union[int, float, Literal['fro', 'nuc']]])r rBrCr!cCs.|jtvrtdttjj|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)r8r*r,rBrC) r"rr#rr(r$r%normr')r rBrCrrr matrix_norms rG)r nr!cCs(|jtvrtdttj|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 rHrrrrIs rI)rtolzOptional[Union[float, Array]])r rJr!cCs|jdkrtjdtjj|jdd}|dur`|jdddt|jd dt|j j }n2t |t rp|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-dimensionalFZ compute_uvNr*T)r,rBr8.r+)r3r$r%Z LinAlgErrorsvdr'maxr1finfor"eps isinstancerasarrayZnewaxisr(Z count_nonzero)r rJrZtolrrr matrix_ranks  0 "rScCs(|jdkrtdtt|jddS)NrKz5x must be at least 2-dimensional for matrix_transposer*r8)r3r2rr(r$Zswapaxesr'r6rrrmatrix_transposes rTcCsN|jtvs|jtvrtd|jdks0|jdkr8tdtt|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#r3r2rr(r$outerr'rArrrrUs rUcCsR|jtvrtd|dur:t|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 pinvNr8)Zrcond) r"rr#rNr1r$rOrPrr(r%pinvr')r rJrrrrVs   rVZreducedmodezLiteral['reduced', 'complete'])r rXr!cCs0|jtvrtdtttjtjj |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 qrrW) r"rr#rr<rr(r$r%qrr')r rXrrrrYs rYcCs,|jtvrtdtttjtj |j S)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#rr<rr(r$r%slogdetr'r6rrrrZ s rZcCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkrx| j }n| j }|| rdnd}||}|||||d}| |j | dd S) NrK) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->d) signatureextobjF)copy)Z linalg.linalgr[r\r]r^r_r`rar%rbr3Zsolve1solveZastype)abr[r\r]r^r_r`rarb_wraptZresult_tZgufuncrcrdrrrr_solve!s$    rmcCs0|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(rmr'rArrrrf<srfT full_matrices)r ror!cCs0|jtvrtdtttjtjj |j |dS)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 svdrn) r"rr#rr<rr(r$r%rMr')r rorrrrMIs rMzUnion[Array, Tuple[Array, ...]]cCs*|jtvrtdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrL) r"rr#rr(r$r%rMr'r6rrrsvdvalsZs rprKaxesz/Union[int, Tuple[Sequence[int], Sequence[int]]])r.r/rrr!cCs6|jtvs|jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotrq)r"rr#rr(r$ tensordotr')r.r/rrrrrrsbsrsc Cs2|jtvrtdtttj|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 tracer8r*r9) r"rr#rr(r$rRtracer'r;rrrrtks rtc Cs|jtvs|jtvrtdt|j|j}d||jt|j}d||jt|j}||||krrtdt |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 axisr*.).N).rr)r"rr#rNr3tupler1r2r$Zbroadcast_arraysr'Zmoveaxisrr() r.r/r,r3Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdotxsrwrEz%Optional[Union[int, Tuple[int, ...]]]zOptional[Union[int, float]])r r,rBrCr!cs|jtvrtd|jdur.dnnttrtfddtjD}|}t | t fddDgfdd|DRdt t jj||dS) rDz.Only floating-point dtypes are allowed in normNrc3s|]}|vr|VqdS)Nr.0ir+rr zvector_norm..csg|]}j|qSr)r1rx)rgrr r|zvector_norm..rE)r"rr#r'flattenrQruranger3r$Z transposeZreshapeprodrr(r%rF)r r,rBrCrestZnewshaper)rgr,r vector_norms  <r)r&r4r5r:r=r>r?r@rGrIrSrTrUrVrYrZrfrMrprsrtrwr)0 __future__rZ_dtypesrrZ _array_objectrtypingrZ_typingrr r r r r Z numpy.linalgnumpyr$rrrrr&r4r5r:r=r>r?r@rGrIrSrTrUrVrYrZrmrfrMrprsrtrwr__all__rrrrsJ