o hE@sddlmZddlmZmZddlmZddlmZddl m Z ddl m Z e r6dd l mZmZmZmZmZdd l mZdd lZdd lZGd d d eZGdddeZGdddeZGdddeZdddkddZdddld#d$Zdmd%d&Zdd'dnd)d*Zdod+d,Zdmd-d.Z dmd/d0Z!dpd1d2Z"dd3d4dqd8d9Z#drd;d<Z$d d=dsd@dAZ%dmdBdCZ&dpdDdEZ'd d=dsdFdGZ(dHdIdtdLdMZ)dudNdOZ*dPdQZ+dpdRdSZ,dTdUdvdWdXZ-dwdZd[Z.dd\dxd_d`Z/dd'dndadbZ0dddldcddZ1d dddedydhdiZ2gdjZ3d S)z) annotations)_floating_dtypes_numeric_dtypes)reshape)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc@eZdZUded<ded<dS) EighResultr eigenvalues eigenvectorsN__name__ __module__ __qualname____annotations__rrQ/var/www/html/scripts/venv/lib/python3.10/site-packages/numpy/array_api/linalg.pyr  rc@r)QRResultrQRNrrrrrrrrc@r) SlogdetResultrsign logabsdetNrrrrrr rr c@s&eZdZUded<ded<ded<dS) SVDResultrUSVhNrrrrrr#s  r#F)upperxrr'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'Lrrrr/%s   r/axisx1x2r6intcCsr|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 3r5) r+rr,shape ValueErrorndimrr1r-crossr0)r7r8r6rrrr>5s  r>cC&|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) r+rr,rr1r-r.detr0r(rrrr@Gs r@)offsetrBcCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r4rBaxis1axis2)rr1r-diagonalr0r(rBrrrrGTsrGcC,|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) r+rr,rmaprr1r-r.eighr0rArrrrK_ rKcCr?)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,rr1r-r.eigvalshr0rArrrrMo rMcCr?)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,rr1r-r.invr0rArrrrO|rNrOcCs2|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,rr1r-matmulr0r7r8rrrrPsrPfro)keepdimsordrSrT2Optional[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)rCr4r6rSrT) r+rr,rr1r-r.normr0)r(rSrTrrr matrix_norms rYncCs(|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) r+rr,rr1r-r. matrix_powerr0)r(rZrrrr[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. rzA1-dimensional array given. Array must be at least two-dimensionalF compute_uvNr4T)r6rSrC.r5)r=r-r. LinAlgErrorsvdr0maxr;finfor+eps isinstancerasarraynewaxisr1 count_nonzero)r(r\r%tolrrr matrix_ranks  0 "rjcCs(|jdkr tdtt|jddS)Nrz5x must be at least 2-dimensional for matrix_transposer4rC)r=r<rr1r-swapaxesr0rArrrmatrix_transposes rlcCsN|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) r+rr,r=r<rr1r-outerr0rQrrrrms rmcCsR|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 pinvNrC)rcond) r+rr,rbr;r-rcrdrr1r.pinvr0)r(r\rrrros  roreducedmoderrLiteral['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 qrrq) r+rr,rrJrr1r-r.qrr0)r(rrrrrru rucCrI)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 rJrr1r-r.slogdetr0rArrrrw rLrwcCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkr<| j }n| j }|| rEdnd}||}|||||d}| |j | dd S) Nr) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->d) signatureextobjF)copy) linalg.linalgrxryrzr{r|r}r~r.rr=solve1solveastype)abrxryrzr{r|r}r~r_wraptresult_tgufuncrrrrrr_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)r+rr,rr1rr0rQrrrr?srT full_matricesrcCrt)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) r+rr,r#rJrr1r-r.rar0)r(rrrrraLrvraUnion[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFr^) r+rr,rr1r-r.rar0rArrrsvdvals]s raxesr/Union[int, Tuple[Sequence[int], Sequence[int]]]cCs6|jtvs |jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr)r+rr,rr1r- tensordotr0)r7r8rrrrresrc Cs2|jtvr tdtttj|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 tracerCr4rD) r+rr,rr1r-rftracer0rHrrrrns  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 axisr4.).N).rr)r+rr,rbr=tupler;r<r-broadcast_arraysr0moveaxisrr1) r7r8r6r=x1_shapex2_shapex1_x2_resrrrvecdot{srrW%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) rVz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr.0i)normalized_axisrr szvector_norm..csg|]}j|qSr)r;r)rrr szvector_norm..)r+)r6rTr)r+rr,r0ravelrerr r=ranger- transposerprodr9rr1r.rXlistr;) r(r6rSrT_axisrestnewshaperr;rr)rrr vector_norms.    0  r)r/r>r@rGrKrMrOrPrYr[rjrlrmrorurwrrarrrrr)r(rr'r)r*r)r7rr8rr6r9r*r)r(rr*r)r(rrBr9r*r)r(rr*r)r7rr8rr*r)r(rrSr)rTrUr*r)r(rrZr9r*r)r(rr\r]r*r)r(rrrrsr*r)r(rr*r )r(rrr)r*r#)r(rr*r)r7rr8rrrr*r) r(rr6rrSr)rTrr*r)4 __future__r_dtypesrr_manipulation_functionsr _array_objectr core.numericr typingr _typingr r r rrr numpy.linalgnumpyr-rrr r#r/r>r@rGrKrMrOrPrYr[rjrlrmrorurwrrrarrrrr__all__rrrrsP                  -