o emG@s&ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZer@dd lmZmZmZmZmZmZdd lmZdd lZdd lZGd d d eZGdddeZGdddeZGdddeZdddnddZ dddod#d$Z!dpd%d&Z"dd'dqd)d*Z#drd+d,Z$dpd-d.Z%dpd/d0Z&dsd1d2Z'dd3d4dtd8d9Z(dud;d<Z)d d=dvd@dAZ*dpdBdCZ+dsdDdEZ,d d=dvdFdGZ-dHdIdwdLdMZ.dxdNdOZ/dPdQZ0dsdRdSZ1dTdUdydWdXZ2dzdZd[Z3dd\d{d_d`Z4dd dad|dddeZ5dddodfdgZ6d dddhd}dkdlZ7gdmZ8d S)~) annotations)_floating_dtypes_numeric_dtypesfloat32float64 complex64 complex128)reshape)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnionDtype) NamedTupleNc@eZdZUded<ded<dS) EighResultr eigenvalues eigenvectorsN__name__ __module__ __qualname____annotations__rrI/var/www/html/venv/lib/python3.10/site-packages/numpy/array_api/linalg.pyr  rc@r)QRResultr QRNrrrrr r"r!r"c@r) SlogdetResultr sign logabsdetNrrrrr r%!r!r%c@s&eZdZUded<ded<ded<dS) SVDResultr USVhNrrrrr r(%s  r(F)upperxr r,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,Lrrr r4,s   r4axisx1x2r;intcCsr|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 3r:) r0rr1shape ValueErrorndimr r6r2crossr5)r<r=r;rrr rC<s  rCcC&|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) r0rr1r r6r2r3detr5r-rrr rENs rE)offsetrGcCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r9)rGaxis1axis2)r r6r2diagonalr5)r-rGrrr rK[srKcC,|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) r0rr1rmapr r6r2r3eighr5rFrrr rNf rNcCrD)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) r0rr1r r6r2r3eigvalshr5rFrrr rPv rPcCrD)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) r0rr1r r6r2r3invr5rFrrr rRrQrRcCs2|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)r0rr1r r6r2matmulr5r<r=rrr rSsrSfro)keepdimsordrVrW2Optional[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)rHr9r;rVrW) r0rr1r r6r2r3normr5)r-rVrWrrr matrix_norms r\ncCs(|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) r0rr1r r6r2r3 matrix_powerr5)r-r]rrr r^s r^)rtolr_Optional[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. r zA1-dimensional array given. Array must be at least two-dimensionalF compute_uvNr9T)r;rVrH.r:)rBr2r3 LinAlgErrorsvdr5maxr@finfor0eps isinstancer asarraynewaxisr6 count_nonzero)r-r_r*tolrrr matrix_ranks  0 "rmcCs(|jdkr tdtt|jddS)Nr z5x must be at least 2-dimensional for matrix_transposer9rH)rBrAr r6r2swapaxesr5rFrrr matrix_transposes rocCsN|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) r0rr1rBrAr r6r2outerr5rTrrr rps rpcCsR|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 pinvNrH)rcond) r0rr1rer@r2rfrgr r6r3pinvr5)r-r_rrr rrs  rrreducedmoderuLiteral['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 qrrt) r0rr1r"rMr r6r2r3qrr5)r-rurrr rx rxcCrL)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) r0rr1r%rMr r6r2r3slogdetr5rFrrr rzrOrzcCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkr<| j }n| j }|| rEdnd}t j|ddddd ||||d }Wdn1scwY| |j| d d S) Nr ) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->dcallignore)rinvalidoverdivideunder) signatureF)copy) linalg.linalgr{r|r}r~rrrr3rrBsolve1solver2errstateastype)abr{r|r}r~rrrr_wraptresult_tgufuncrrrrr _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)r0rr1r r6rr5rTrrr rGsrT full_matricesrcCrw)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) r0rr1r(rMr r6r2r3rdr5)r-rrrr rdTryrdUnion[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFra) r0rr1r r6r2r3rdr5rFrrr svdvalses raxesr/Union[int, Tuple[Sequence[int], Sequence[int]]]cCs6|jtvs |jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr)r0rr1r r6r2 tensordotr5)r<r=rrrr rmsr)rGr0r0Optional[Dtype]c CsZ|jtvr td|dur|jtkrt}n|jtkrt}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 traceNrHr9)rGrIrJr0) r0rr1rrrr r r6r2ritracer5)r-rGr0rrr rvs   "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 axisr9.).N).rr)r0rr1rerBtupler@rAr2broadcast_arraysr5moveaxisr r6) r<r=r;rBx1_shapex2_shapex1_x2_resrrr vecdotsrrZ%Optional[Union[int, Tuple[int, ...]]]Optional[Union[int, float]]c s|jtvr td|j|durd}nBt|trWt||jtfddt jD}||}t | t j fdd|Dtdgfdd|DRd}n|}tt jj||d }|rt|j}t|durvt |jn||j}|D]} d || <q}t |t|}|S) rYz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr.0i)normalized_axisrr szvector_norm..csg|]}j|qSr)r@r)rrr szvector_norm..)r0)r;rWr)r0rr1r5ravelrhrr rBranger2 transposer prodr>r r6r3r[listr@) r-r;rVrW_axisrestnewshaperr@rr)rrr vector_norms.    0  r)r4rCrErKrNrPrRrSr\r^rmrorprrrxrzrrdrrrrr)r-r r,r.r/r )r<r r=r r;r>r/r )r-r r/r )r-r rGr>r/r )r-r r/r)r<r r=r r/r )r-r rVr.rWrXr/r )r-r r]r>r/r )r-r r_r`r/r )r-r rurvr/r")r-r r/r%)r-r rr.r/r()r-r r/r)r<r r=r rrr/r )r-r rGr>r0rr/r ) r-r r;rrVr.rWrr/r )9 __future__r_dtypesrrrrrr _manipulation_functionsr _array_objectr core.numericr typingr_typingrrrrrrr numpy.linalgnumpyr2rr"r%r(r4rCrErKrNrPrRrSr\r^rmrorprrrxrzrrrdrrrrr__all__rrrr sP                  -