o ҷhG@s2ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZerFdd lmZmZmZmZmZmZdd lmZdd lZdd lZGd ddeZGdddeZGdddeZ GdddeZ!dddoddZ"dddpd$d%Z#dqd&d'Z$dd(drd*d+Z%dsd,d-Z&dqd.d/Z'dqd0d1Z(dtd2d3Z)dd4d5dud9d:Z*dvddwdAdBZ,dqdCdDZ-dtdEdFZ.d d>dwdGdHZ/dIdJdxdMdNZ0dydOdPZ1dQdRZ2dtdSdTZ3dUdVdzdXdYZ4d{d[d\Z5dd]d|d`daZ6dd dbd}dedfZ7dddpdgdhZ8d dddid~dldmZ9gdnZ:d S)) annotations)_floating_dtypes_numeric_dtypesfloat32float64 complex64 complex128)reshape)conj)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnionDtype) NamedTupleNc@eZdZUded<ded<dS) EighResultr eigenvalues eigenvectorsN__name__ __module__ __qualname____annotations__r r I/var/www/html/venv/lib/python3.10/site-packages/numpy/array_api/linalg.pyr  rc@r)QRResultr QRNrr r r r!r#r"r#c@r) SlogdetResultr sign logabsdetNrr r r r!r&"r"r&c@s&eZdZUded<ded<ded<dS) SVDResultr USVhNrr r r r!r)&s  r)F)upperxr r-boolreturncCsT|jtvr tdtj|j}|r%t|j }|jt t fvr#t |}|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 _newmTrr r )r.r-Lr*r r r!r5-s   r5axisx1x2r<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;) r1rr2shape ValueErrorndimr r7r3crossr6)r=r>r<r r r!rD@s  rDcC&|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) r1rr2r r7r3r4detr6r.r r r!rFRs rF)offsetrHcCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r:)rHaxis1axis2)r r7r3diagonalr6)r.rHr r r!rL_srLcC,|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) r1rr2rmapr r7r3r4eighr6rGr r r!rOj rOcCrE)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) r1rr2r r7r3r4eigvalshr6rGr r r!rQz rQcCrE)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) r1rr2r r7r3r4invr6rGr r r!rSrRrScCs2|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)r1rr2r r7r3matmulr6r=r>r r r!rTsrTfro)keepdimsordrWrX2Optional[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)rIr:r<rWrX) r1rr2r r7r3r4normr6)r.rWrXr r r! 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) r1rr2r r7r3r4 matrix_powerr6)r.r^r r r!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_uvNr:T)r<rWrI.r;)rCr3r4 LinAlgErrorsvdr6maxrAfinfor1eps isinstancer asarraynewaxisr7 count_nonzero)r.r`r+tolr r r! matrix_ranks  0 "rncCs(|jdkr tdtt|jddS)Nr z5x must be at least 2-dimensional for matrix_transposer:rI)rCrBr r7r3swapaxesr6rGr r r!matrix_transposes rpcCsN|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) r1rr2rCrBr r7r3outerr6rUr r r!rqs rqcCsR|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 pinvNrI)rcond) r1rr2rfrAr3rgrhr r7r4pinvr6)r.r`r r r!rss  rsreducedmodervLiteral['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 qrru) r1rr2r#rNr r7r3r4qrr6)r.rvr r r!ry rycCrM)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) r1rr2r&rNr r7r3r4slogdetr6rGr r r!r{rPr{cCsddlm}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~rrrrr4rrCsolve1solver3errstateastype)abr|r}r~rrrrr_wraptresult_tgufuncrrr r r!_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)r1rr2r r7rr6rUr r r!rKsrT full_matricesrcCrx)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) r1rr2r)rNr r7r3r4rer6)r.rr r r!reXrzreUnion[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrb) r1rr2r r7r3r4rer6rGr r r!svdvalsis raxesr/Union[int, Tuple[Sequence[int], Sequence[int]]]cCs6|jtvs |jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr)r1rr2r r7r3 tensordotr6)r=r>rr r r!rqsr)rHr1r1Optional[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 traceNrIr:)rHrJrKr1) r1rr2rrrr r r7r3rjtracer6)r.rHr1r r r!rzs   "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 axisr:.).N).rr)r1rr2rfrCtuplerArBr3broadcast_arraysr6moveaxisr r7) r=r>r<rCx1_shapex2_shapex1_x2_resr r r!vecdotsrr[%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) rZz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr .0i)normalized_axisr r! szvector_norm..csg|]}j|qSr )rAr)rr r! szvector_norm..)r1)r<rXr)r1rr2r6ravelrirrrCranger3 transposer prodr?r r7r4r\listrA) r.r<rWrX_axisrestnewshaperrArr )rrr! vector_norms.    0  r)r5rDrFrLrOrQrSrTr]r_rnrprqrsryr{rrerrrrr)r.r r-r/r0r )r=r r>r r<r?r0r )r.r r0r )r.r rHr?r0r )r.r r0r)r=r r>r r0r )r.r rWr/rXrYr0r )r.r r^r?r0r )r.r r`rar0r )r.r rvrwr0r#)r.r r0r&)r.r rr/r0r))r.r r0r)r=r r>r rrr0r )r.r rHr?r1rr0r ) r.r r<rrWr/rXrr0r ); __future__r_dtypesrrrrrr _manipulation_functionsr _elementwise_functionsr _array_objectr core.numericrtypingr_typingrrrrrrr numpy.linalgnumpyr3rr#r&r)r5rDrFrLrOrQrSrTr]r_rnrprqrsryr{rrrerrrrr__all__r r r r!sR                   -