o &iG@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) EighResultrZ eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrAC:\wamp64\www\opt\env\Lib\site-packages\numpy/array_api/linalg.pyr  rc@r)QRResultrQRNrrrrrrrrc@r) SlogdetResultrsignZ logabsdetNrrrrrrrrc@s&eZdZUded<ded<ded<dS) SVDResultrUSZVhNrrrrrr 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_newZmT)r$r#Lrrrr+%s   r+axisx1x2r1intcCsr|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 3r0) r'rr(shape ValueErrorndimrr-r)crossr,)r2r3r1rrrr95s  r9cC&|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(rr-r)r*detr,r$rrrr;Gs r;)offsetr=cCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r/r=Zaxis1Zaxis2)rr-r)diagonalr,r$r=rrrr@Tsr@cC,|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(rmaprr-r)r*eighr,r<rrrrD_ rDcCr:)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,r<rrrrFo rFcCr:)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,r<rrrrH|rGrHcCs2|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(rr-r)matmulr,r2r3rrrrIsrIZfro)keepdimsordrKrL2Optional[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)r>r/r1rKrL) r'rr(rr-r)r*normr,)r$rKrLrrr matrix_norms rQncCs(|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(rr-r)r* matrix_powerr,)r$rRrrrrSs rS)rtolrTOptional[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-dimensionalFZ compute_uvNr/T)r1rKr>.r0)r8r)r*Z LinAlgErrorsvdr,maxr6finfor'eps isinstancerasarrayZnewaxisr-Z count_nonzero)r$rTr"Ztolrrr matrix_ranks  0 "r]cCs(|jdkr tdtt|jddS)Nrz5x must be at least 2-dimensional for matrix_transposer/r>)r8r7rr-r)Zswapaxesr,r<rrrmatrix_transposes r^cCsN|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(r8r7rr-r)outerr,rJrrrr_s r_cCsR|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 pinvNr>)Zrcond) r'rr(rXr6r)rYrZrr-r*pinvr,)r$rTrrrr`s  r`ZreducedmoderbLiteral['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 qrra) r'rr(rrCrr-r)r*qrr,)r$rbrrrre recCrB)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(rrCrr-r)r*slogdetr,r<rrrrg rErgcCsddlm}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)Z linalg.linalgrhrirjrkrlrmrnr*ror8Zsolve1solveZastype)abrhrirjrkrlrmrnro_wraptZresult_tZgufuncrprqrrrr_solve$s$    rzcCs0|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(rr-rzr,rJrrrrs?srsT full_matricesr|cCrd)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 rCrr-r)r*rWr,)r$r|rrrrWLrfrWUnion[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrV) r'rr(rr-r)r*rWr,r<rrrsvdvals]s r~axesr/Union[int, Tuple[Sequence[int], Sequence[int]]]cCs6|jtvs |jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr)r'rr(rr-r) tensordotr,)r2r3rrrrresrc 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 tracer>r/r?) r'rr(rr-r)r\tracer,rArrrrns  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)r'rr(rXr8tupler6r7r)Zbroadcast_arraysr,Zmoveaxisrr-) r2r3r1r8Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdot{srrO%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) rNz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr.0i)normalized_axisrr szvector_norm..csg|]}j|qSr)r6r)rtrr szvector_norm..)r')r1rLr)r'rr(r,Zravelr[rr r8ranger)Z transposerprodr4rr-r*rPlistr6) r$r1rKrLZ_axisrestZnewshaperr6rr)rtrr vector_norms.    0  r)r+r9r;r@rDrFrHrIrQrSr]r^r_r`rergrsrWr~rrrr)r$rr#r%r&r)r2rr3rr1r4r&r)r$rr&r)r$rr=r4r&r)r$rr&r)r2rr3rr&r)r$rrKr%rLrMr&r)r$rrRr4r&r)r$rrTrUr&r)r$rrbrcr&r)r$rr&r)r$rr|r%r&r )r$rr&r})r2rr3rrrr&r) r$rr1rrKr%rLrr&r)4 __future__rZ_dtypesrrZ_manipulation_functionsrZ _array_objectrZ core.numericr typingr Z_typingr r r rrrZ numpy.linalgnumpyr)rrrr r+r9r;r@rDrFrHrIrQrSr]r^r_r`rergrzrsrWr~rrrr__all__rrrrsP                  -