U b@@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.ddd d$d&d(d*d-d3d6d;d=d?dAdGdIdMdRdUd[d]d_degZ/dS)f) annotations)_floating_dtypes_numeric_dtypes)Array) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc@seZdZUded<ded<dS) EighResultrZ eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rr:/tmp/pip-unpacked-wheel-cxuynqjx/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:|jtkrtdtj|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|jtks|jtkrtd|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&|jtkrtdttj|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 rrrr6Ds r6)offset)r r8r!cCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r*r8Zaxis1Zaxis2)rr(r$diagonalr'r r8rrrr;Qsr;cCs,|jtkrtdtttjtj |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'r7rrrr>\s r>cCs&|jtkrtdttj|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'r7rrrr?ls r?cCs&|jtkrtdttj|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'r7rrrr@ys r@)r.r/r!cCs2|jtks|jtkrtdtt|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/rrrrAsrAZfro)keepdimsordz4Optional[Union[int, float, Literal[('fro', 'nuc')]]])r rCrDr!cCs.|jtkrtdttjj|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)r9r*r,rCrD) r"rr#rr(r$r%normr')r rCrDrrr matrix_norms rH)r nr!cCs(|jtkrtdttj|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 rIrrrrJs rJ)rtolzOptional[Union[float, Array]])r rKr!cCs|jdkrtjdtjj|jdd}|dkr`|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,rCr9.r+)r3r$r%Z LinAlgErrorsvdr'maxr1finfor"eps isinstancerasarrayZnewaxisr(Z count_nonzero)r rKrZtolrrr matrix_ranks  0 "rTcCs(|jdkrtdtt|jddS)NrLz5x must be at least 2-dimensional for matrix_transposer*r9)r3r2rr(r$Zswapaxesr'r7rrrmatrix_transposes rUcCsN|jtks|jtkrtd|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'rBrrrrVs rVcCsR|jtkrtd|dkr: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 pinvNr9)Zrcond) r"rr#rOr1r$rPrQrr(r%pinvr')r rKrrrrWs   rWZreducedmodez Literal[('reduced', 'complete')])r rYr!cCs0|jtkrtdtttjtjj |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 qrrX) r"rr#rr=rr(r$r%qrr')r rYrrrrZs rZcCs,|jtkrtdtttjtj |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'r7rrrr[ s r[cCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkrx| j }n| j }|| rdnd}||}|||||d}| |j | dd S) NrL) _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`rarbr%rcr3Zsolve1solveZastype)abr\r]r^r_r`rarbrc_wraptZresult_tZgufuncrdrerrrr_solve!s$    rncCs0|jtks|jtkrtdtt|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(rnr'rBrrrrg<srgT full_matrices)r rpr!cCs0|jtkrtdtttjtjj |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 svdro) r"rr#rr=rr(r$r%rNr')r rprrrrNIs rNzUnion[Array, Tuple[Array, ...]]cCs*|jtkrtdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrM) r"rr#rr(r$r%rNr'r7rrrsvdvalsZs rqrLaxesz/Union[int, Tuple[Sequence[int], Sequence[int]]])r.r/rsr!cCs6|jtks|jtkrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotrr)r"rr#rr(r$ tensordotr')r.r/rsrrrrtbsrtc Cs2|jtkrtdtttj|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 tracer9r*r:) r"rr#rr(r$rStracer'r<rrrruks rucCs2|jtks|jtkrtdt|||f|ffdS)Nz)Only numeric dtypes are allowed in vecdotrr)r"rr#rtr5rrrvecdotxsrvrFz%Optional[Union[int, Tuple[int, ...]]]zOptional[Union[int, float]])r r,rCrDr!cs|jtkrtd|jdkr.dnlttrtfddtjD}|}t | t fddDffdd|Ddt t jj||dS) rEz.Only floating-point dtypes are allowed in normNrc3s|]}|kr|VqdS)Nr.0ir+rr szvector_norm..csg|]}j|qSr)r1rw)rhrr szvector_norm..rF)r"rr#r'flattenrRtupleranger3r$Z transposeZreshapeprodrr(r%rG)r r,rCrDrestZnewshaper)rhr,r vector_norms  :r)0 __future__rZ_dtypesrrZ _array_objectrtypingr_typingrr r r r r Z numpy.linalgZnumpyr$rrrrr&r4r6r;r>r?r@rArHrJrTrUrVrWrZr[rnrgrNrqrtrurvr__all__rrrrsJ