cEddlmZddlmZmZddlmZddlmZddl m Z ddl m Z e rdd l mZmZmZmZmZdd l mZdd lZdd lZGd d eZGddeZGddeZGddeZdddSdZdddTd"ZdUd#Zdd$dVd&ZdWd'ZdUd(Z dUd)Z!dXd*Z"dd+d,dYd0Z#dZd2Z$d d3d[d6Z%dUd7Z&dXd8Z'd d3d[d9Z(d:d;d\d>Z)d]d?Z*d@Z+dXdAZ,dBdCd^dEZ-d_dGZ.ddHd`dKZ/dd$dVdLZ0dddTdMZ1d dddNdadQZ2gdRZ3d S)b) annotations)_floating_dtypes_numeric_dtypes)reshape)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc$eZdZUded<ded<dS) EighResultr eigenvalues eigenvectorsN__name__ __module__ __qualname____annotations__8/usr/lib/python3/dist-packages/numpy/array_api/linalg.pyrrs*rrc$eZdZUded<ded<dS)QRResultrQRNrrrrrrs" HHH HHHHHrrc$eZdZUded<ded<dS) SlogdetResultrsign logabsdetNrrrrr#r#s'KKKrr#c.eZdZUded<ded<ded<dS) SVDResultrUSVhNrrrrr'r's+ HHH HHH IIIIIrr'F)upperxrr+boolreturnc|jtvrtdtj|j}|rtj|j Stj|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+Ls rr4r4%s_ w&&&LMMM 18$$A  z!}} :a==raxisx1x2r;intcp|jtvs|jtvrtd|j|jkrt d|jdkrt d|j|dkrt dt jtj |j |j |S)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 ValueErrorndimrr6r2crossr5)r<r=r;s rrDrD5s  x&&"(/*I*IBCCC x28=>>> w!||JKKK x~9::: :bhry")$??? @ @@rc|jtvrtdtjt j|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) r0rr1rr6r2r3detr5r,s rrFrFGsA w&&&GHHH :bimmAH-- . ..r)offsetrHc`tjtj|j|ddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r9rHaxis1axis2)rr6r2diagonalr5r,rHs rrNrNTs) :bk!(62NNN O OOrc|jtvrtdtt t jtj |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) r0rr1rmaprr6r2r3eighr5rGs rrRrR_sJ w&&&HIII s5:ry~~ah'?'?@@ AArc|jtvrtdtjt j|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) r0rr1rr6r2r3eigvalshr5rGs rrTrTosC w&&&LMMM :bi((22 3 33rc|jtvrtdtjt j|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) r0rr1rr6r2r3invr5rGs rrVrV|sA w&&&GHHH :bimmAH-- . ..rc|jtvs|jtvrtdtjt j|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)r0rr1rr6r2matmulr5r<r=s rrXrXsL x&&"(/*I*ICDDD :bi 2955 6 66rfro)keepdimsordr[r\2Optional[Union[int, float, Literal['fro', 'nuc']]]c|jtvrtdtjt j|jd||S) 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)rJr9r;r[r\) r0rr1rr6r2r3normr5)r,r[r\s r matrix_normrbsJ w&&&OPPP :binnQXHxUXnYY Z ZZrnc|jtvrtdtjt j|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) r0rr1rr6r2r3 matrix_powerr5)r,rcs rreresG w&&&ghhh :bi,,QXq99 : ::r)rtolrfOptional[Union[float, Array]]cZ|jdkrtjdtj|jd}|T|ddt |jd dztj|j j z}nZt|tr|j}|ddtj |d tjfz}tjtj||kd 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;r[rJ.r:)rCr2r3 LinAlgErrorsvdr5maxrAfinfor0eps isinstancerasarraynewaxisr6 count_nonzero)r,rfr)tols r matrix_rankrus vzzi##$ghhh ah5 11A |eede++c!'"##,.?.??"(17BSBSBWW dE " " ;Deede++BJt,<,`. See its docstring for more information. z(Only numeric dtypes are allowed in outerrz/The input arrays to outer must be 1-dimensional) r0rr1rCrBrr6r2outerr5rYs rrzrzsr x&&"(/*I*IBCCC w!||rw!||JKKK :bhry")44 5 55rc&|jtvrtd|;t|jddt j|jjz}tj t j |j |S)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 pinvNrJ)rcond) r0rr1rmrAr2rnrorr6r3pinvr5)r,rfs rr}r}sw w&&&HIII |17233<  28AG#4#4#88 :binnQXTn:: ; ;;rreducedmoderLiteral['reduced', 'complete']c |jtvrtdtt t jtj |j |S)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) r0rr1rrQrr6r2r3qrr5)r,rs rrrsN w&&&FGGG SRY\\!(\%F%FGG HHrc|jtvrtdtt t jtj |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) r0rr1r#rQrr6r2r3slogdetr5rGs rrr sL w&&&KLLL #ej")*;*;AH*E*EFF GGrczddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkr| j }n| j }|| rdnd}||}|||||}| | | d S) Nr ) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->d) signatureextobjF)copy) linalg.linalgrrrrrrrr3rrCsolve1solveastype)abrrrrrrrr_wraptresult_tgufuncrrrs r_solver$sK>>>>>>>>>>>>>>>>>>'&&&&& :a==DAqq1jmmGAt+a##KAx v{{%$)=++8I $ $%@ A AFq!y888A 4.. / //rc|jtvs|jtvrtdtjt |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)r0rr1rr6rr5rYs rrr?sL x'''28;K+K+KIJJJ :fRY 22 3 33rT full_matricesrc |jtvrtdtt t jtj |j |S)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'rQrr6r2r3rlr5)r,rs rrlrlLsN w&&&GHHH c%*bimmAHMm&Z&Z[[ \\rUnion[Array, Tuple[Array, ...]]c|jtvrtdtjt j|jdS)Nz1Only floating-point dtypes are allowed in svdvalsFri) r0rr1rr6r2r3rlr5rGs rsvdvalsr]sCw&&&KLLL :bimmAHm?? @ @@raxesr/Union[int, Tuple[Sequence[int], Sequence[int]]]c|jtvs|jtvrtdtjt j|j|j|S)Nz,Only numeric dtypes are allowed in tensordotr)r0rr1rr6r2 tensordotr5)r<r=rs rrresQ x&&"(/*I*IFGGG :bl29bidCCC D DDrc |jtvrtdtjt jt j|j|ddS)zz Array API compatible wrapper for :py:func:`np.trace `. See its docstring for more information. z(Only numeric dtypes are allowed in tracerJr9rK) r0rr1rr6r2rqtracer5rOs rrrnsR  wo%%BCCC :bj!(6SU!V!V!VWW X XXrcR|jtvs|jtvrtdt|j|j}d||jz zt |jz}d||jz zt |jz}||||krtdtj |j |j \}}tj ||d}tj ||d}|ddddf|dz}tj |dS)Nz)Only numeric dtypes are allowed in vecdot)rz6x1 and x2 must have the same size along the given axisr9.).N).rr)r0rr1rmrCtuplerArBr2broadcast_arraysr5moveaxisrr6) r<r=r;rCx1_shapex2_shapex1_x2_ress rvecdotr{s  x&&"(/*I*ICDDD rw DTBG^$uRX6HTBG^$uRX6H~$''QRRR"29bi88HC +c4 $ $C +c4 $ $C c4l c)n ,C :c)n % %%rr`%Optional[Union[int, Tuple[int, ...]]]Optional[Union[int, float]]c8 |jtvrtd|j |  d}nt |t rt||j t fdt jD}||z}tj | tj fd|Dtg fd|DR d}n|}tjtj ||}|rft'|j}t|t|jn||j}|D]} d || <t|t |}|S) r_z.Only floating-point dtypes are allowed in normNrc3$K|] }|v|V dS)Nr).0inormalized_axiss r zvector_norm..s-JJ1/1I1IQ1I1I1I1IJJrc*g|]}j|SrrArrrs r zvector_norm..s///Qagaj///r)r0c*g|]}j|Srrrs rrzvector_norm..s>X>X>Xaqwqz>X>X>Xr)r;r\r)r0rr1r5ravelrprr rCranger2 transposerprodr>rr6r3ralistrA) r,r;r[r\_axisrestnewshaperrArrrs @@r vector_normrs w&&&HIII A | GGII D% /tQV<<JJJJaf JJJJJ$; LH % % - - W////$///s ; ; ; Y>X>X>X>XSW>X>X>X Y Y [ [ *RY^^AEs^;; < r.r)r,rr.r)r,rrHr>r.r)r,rr.r)r<rr=rr.r)r,rr[r-r\r]r.r)r,rrcr>r.r)r,rrfrgr.r)r,rrrr.r)r,rr.r#)r,rrr-r.r')r,rr.r)r<rr=rrrr.r) r,rr;rr[r-r\rr.r)4 __future__r_dtypesrr_manipulation_functionsr _array_objectr core.numericr typingr _typingr r rrrr numpy.linalgnumpyr2rrr#r'r4rDrFrNrRrTrVrXrbrerurxrzr}rrrrrlrrrrr__all__rrrrsp""""""66666666,,,,,, ////// CBBBBBBBBBBBBBB     z   J ,1       35AAAAAA$ / / / /,-PPPPPP B B B B 4 4 4 4 / / / / 7 7 7 7(27rw [ [ [ [ [ [ ; ; ; ;GK::::::05555 6666$@D<<<<<<"?H I I I I I I H H H H00006 4 4 4 4/3 ] ] ] ] ] ]"AAAAcdEEEEEE)* Y Y Y Y Y Y46&&&&&&,OSejPQ++++++Z F F Fr