a ;emG@sddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZerdd lmZmZmZmZmZmZdd lmZdd lZdd lZGd d d eZGdddeZGdddeZGdddeZddddddddZ ddddddddd Z!ddd!d"d#Z"dd$dddd%d&d'Z#dd d!d(d)Z$ddd!d*d+Z%ddd!d,d-Z&dddd.d/d0Z'dd1d2ddd3dd4d5d6Z(dddd7d8d9Z)d d:dd;dd<d=d>Z*ddd!d?d@Z+dddd.dAdBZ,d d:dd;dd<dCdDZ-dEdFddGddHdIdJZ.ddd!dKdLZ/dMdNZ0dddd.dOdPZ1dQdRddddSdTdUZ2ddVd!dWdXZ3ddYdddZdd[d\d]Z4dd d^ddd_dd`dadbZ5ddddddddcddZ6d dddeddfddgddhdidjZ7gdkZ8d S)l) annotations)_floating_dtypes_numeric_dtypesfloat32float64 complex64 complex128)reshape)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnionDtype) NamedTupleNc@seZdZUded<ded<dS) EighResultr Z eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrJ/var/www/sunrise/env/lib/python3.9/site-packages/numpy/array_api/linalg.pyrs rc@seZdZUded<ded<dS)QRResultr QRNrrrrrrs rc@seZdZUded<ded<dS) SlogdetResultr signZ logabsdetNrrrrrr!!s r!c@s&eZdZUded<ded<ded<dS) SVDResultr USZVhNrrrrrr#%s r#F)upperr bool)xr&returncCs:|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(r&Lrrrr.,s   r.axisint)x1x2r4r)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 3r3) r*rr+shape ValueErrorndimr r0r,crossr/)r6r7r4rrrr<<s  r<)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+r r0r,r-detr/r(rrrr=Ns r=)offset)r(r?r)cCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r2)r?axis1axis2)r r0r,diagonalr/)r(r?rrrrC[srCcCs,|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+rmapr r0r,r-eighr/r>rrrrEfs rEcCs&|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+r r0r,r-eigvalshr/r>rrrrFvs rFcCs&|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+r r0r,r-invr/r>rrrrGs rG)r6r7r)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+r r0r,matmulr/r6r7rrrrHsrHZfro)keepdimsordz2Optional[Union[int, float, Literal['fro', 'nuc']]])r(rJrKr)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)r@r2r4rJrK) r*rr+r r0r,r-normr/)r(rJrKrrr matrix_norms rO)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+r r0r,r- matrix_powerr/)r(rPrrrrQs rQ)rtolzOptional[Union[float, Array]])r(rRr)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. r zA1-dimensional array given. Array must be at least two-dimensionalFZ compute_uvNr2T)r4rJr@.r3)r;r,r-Z LinAlgErrorsvdr/maxr9finfor*eps isinstancer asarrayZnewaxisr0Z count_nonzero)r(rRr%Ztolrrr matrix_ranks  0 "rZcCs(|jdkrtdtt|jddS)Nr z5x must be at least 2-dimensional for matrix_transposer2r@)r;r:r r0r,Zswapaxesr/r>rrrmatrix_transposes r[cCsN|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+r;r:r r0r,outerr/rIrrrr\s r\cCsR|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 pinvNr@)Zrcond) r*rr+rUr9r,rVrWr r0r-pinvr/)r(rRrrrr]s   r]ZreducedmodezLiteral['reduced', 'complete'])r(r_r)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 qrr^) r*rr+rrDr r0r,r-qrr/)r(r_rrrr`s r`cCs,|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+r!rDr r0r,r-slogdetr/r>rrrras racCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkrx| j }n| j }|| rdnd}t j|ddddd ||||d }Wdn1s0Y| |j| d d S) Nr ) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->dcallignore)rjinvalidZoverdivideZunder) signatureF)copy)Z linalg.linalgrbrcrdrerfrgrhr-rir;Zsolve1solver,ZerrstateZastype)abrbrcrdrerfrgrhri_wraptZresult_tZgufuncrnrrrr_solve+s $    ,rwcCs0|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+r r0rwr/rIrrrrpGsrpT full_matrices)r(ryr)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 svdrx) r*rr+r#rDr r0r,r-rTr/)r(ryrrrrTTs rTzUnion[Array, Tuple[Array, ...]]cCs*|jtvrtdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrS) r*rr+r r0r,r-rTr/r>rrrsvdvalses rzaxesz/Union[int, Tuple[Sequence[int], Sequence[int]]])r6r7r|r)cCs6|jtvs|jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr{)r*rr+r r0r, tensordotr/)r6r7r|rrrr}msr})r?r*zOptional[Dtype])r(r?r*r)c CsZ|jtvrtd|dur8|jtkr*t}n|jtkr8t}tt t j |j |dd|dS)zz Array API compatible wrapper for :py:func:`np.trace `. See its docstring for more information. z(Only numeric dtypes are allowed in traceNr@r2)r?rArBr*) r*rr+rrrr r r0r,rYtracer/)r(r?r*rrrr~vs   r~c 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 axisr2.).N).rr)r*rr+rUr;tupler9r:r,Zbroadcast_arraysr/Zmoveaxisr r0) r6r7r4r;Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdotsrrMz%Optional[Union[int, Tuple[int, ...]]]zOptional[Union[int, float]])r(r4rJrKr)c s|jtvrtd|j|dur.d}nt|trt||jtfddt jD}||}t | t j fdd|Dtdgfdd|DRd}n|}tt jj||d }|rt|j}t|durt |jn||j}|D] } d || <qt |t|}|S) rLz.Only floating-point dtypes are allowed in normNrc3s|]}|vr|VqdS)Nr.0i)normalized_axisrr zvector_norm..csg|]}j|qSr)r9r)rqrr rzvector_norm..)r*)r4rKr)r*rr+r/ZravelrXrr r;ranger,Z transposer prodr5r r0r-rNlistr9) r(r4rJrKZ_axisrestZnewshaperr9rr)rqrr vector_norms.    0  r)r.r<r=rCrErFrGrHrOrQrZr[r\r]r`rarprTrzr}r~rr)9 __future__rZ_dtypesrrrrrr Z_manipulation_functionsr Z _array_objectr Z core.numericr typingr_typingrrrrrrrZ numpy.linalgnumpyr,rrr!r#r.r<r=rCrErFrGrHrOrQrZr[r\r]r`rarwrprTrzr}r~rr__all__rrrrsN              -