NgG4ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZerdd lmZmZmZmZmZmZdd lmZdd lZdd lZGd deZGddeZGddeZ GddeZ!dddWdZ"dddXd#Z#dYd$Z$dd%dZd'Z%d[d(Z&dYd)Z'dYd*Z(d\d+Z)dd,d-d]d1Z*d^d3Z+d d4d_d7Z,dYd8Z-d\d9Z.d d4d_d:Z/d;d<d`d?Z0dad@Z1dAZ2d\dBZ3dCdDdbdFZ4dcdHZ5ddIdddLZ6dd dMdedPZ7dddXdQZ8d dddRdfdUZ9gdVZ:d S)g) 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/var/www/html/ai-engine/env/lib/python3.11/site-packages/numpy/array_api/linalg.pyrrs*r"rc$eZdZUded<ded<dS)QRResultr QRNrr!r"r#r%r%s" HHH HHHHHr"r%c$eZdZUded<ded<dS) SlogdetResultr sign logabsdetNrr!r"r#r)r)"s'KKKr"r)c.eZdZUded<ded<ded<dS) SVDResultr USVhNrr!r"r#r-r-&s+ HHH HHH IIIIIr"r-F)upperxr r1boolreturnc.|jtvrtdtj|j}|r?tj|j }|jttfvrt|}|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 _newmTr r r )r2r1Lr.s r#r:r:-s w&&&LMMM 18$$A  JqMM  7y*- - -QA :a==r"axisx1x2rAintcp|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@) r6rr7shape ValueErrorndimr r<r8crossr;)rBrCrAs r#rJrJ@s  x&&"(/*I*IBCCC x28=>>> w!||JKKK x~9::: :bhry")$??? @ @@r"c|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) r6rr7r r<r8r9detr;r2s r#rLrLRsA w&&&GHHH :bimmAH-- . ..r")offsetrNc`tjtj|j|ddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r?)rNaxis1axis2)r r<r8diagonalr;)r2rNs r#rSrS_s) :bk!(62NNN O OOr"c|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) r6rr7rmapr r<r8r9eighr;rMs r#rVrVjsJ w&&&HIII s5:ry~~ah'?'?@@ AAr"c|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) r6rr7r r<r8r9eigvalshr;rMs r#rXrXzsC w&&&LMMM :bi((22 3 33r"c|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) r6rr7r r<r8r9invr;rMs r#rZrZsA w&&&GHHH :bimmAH-- . ..r"c|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)r6rr7r r<r8matmulr;rBrCs r#r\r\sL x&&"(/*I*ICDDD :bi 2955 6 66r"fro)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)rPr?rAr_r`) r6rr7r r<r8r9normr;)r2r_r`s r# matrix_normrfsJ w&&&OPPP :binnQXHxUXnYY Z ZZr"nc|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) r6rr7r r<r8r9 matrix_powerr;)r2rgs r#ririsG w&&&ghhh :bi,,QXq99 : ::r")rtolrjOptional[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. rzA1-dimensional array given. Array must be at least two-dimensionalF compute_uvNr?T)rAr_rP.r@)rIr8r9 LinAlgErrorsvdr;maxrGfinfor6eps isinstancer asarraynewaxisr< count_nonzero)r2rjr/tols r# matrix_rankrys 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) r6rr7rIrHr r<r8outerr;r]s r#r~r~sr x&&"(/*I*IBCCC w!||rw!||JKKK :bhry")44 5 55r"c&|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 pinvNrP)rcond) r6rr7rqrGr8rrrsr r<r9pinvr;)r2rjs r#rrsw w&&&HIII |17233<  28AG#4#4#88 :binnQXTn:: ; ;;r"reducedmoderLiteral['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) r6rr7r%rUr r<r8r9qrr;)r2rs r#rrsN w&&&FGGG SRY\\!(\%F%FGG HHr"c|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) r6rr7r)rUr r<r8r9slogdetr;rMs r#rrsL w&&&KLLL #ej")*;*;AH*E*EFF GGr"cddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkr| j }n| j }|| rdnd}tj|dddd 5|||| }dddn #1swxYwY| || 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.linalgrrrrrrrr9rrIsolve1solver8errstateastype)abrrrrrrrr_wraptresult_tgufuncrrs r#_solver/s>>>>>>>>>>>>>>>>>>'&&&&& :a==DAqq1jmmGAt+a##KAx v{{%$)=++8I 5v"88 E E E.. F1a9 - - -............... 4.. / //sB77B;>B;c|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)r6rr7r r<rr;r]s r#rrKsL x'''28;K+K+KIJJJ :fRY 22 3 33r"T 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) r6rr7r-rUr r<r8r9rpr;)r2rs r#rprpXsN w&&&GHHH c%*bimmAHMm&Z&Z[[ \\r"Union[Array, Tuple[Array, ...]]c|jtvrtdtjt j|jdS)Nz1Only floating-point dtypes are allowed in svdvalsFrm) r6rr7r r<r8r9rpr;rMs r#svdvalsrisCw&&&KLLL :bimmAHm?? @ @@r"axesr/Union[int, Tuple[Sequence[int], Sequence[int]]]c|jtvs|jtvrtdtjt j|j|j|S)Nz,Only numeric dtypes are allowed in tensordotr)r6rr7r r<r8 tensordotr;)rBrCrs r#rrqsQ x&&"(/*I*IFGGG :bl29bidCCC D DDr")rNr6r6Optional[Dtype]c "|jtvrtd|/|jtkrt}n|jt krt }tjtj tj |j |dd|S)zz Array API compatible wrapper for :py:func:`np.trace `. See its docstring for more information. z(Only numeric dtypes are allowed in traceNrPr?)rNrQrRr6) r6rr7rrr r r r<r8rutracer;)r2rNr6s r#rrzs  wo%%BCCC } 7g  EE W ! !E :bj!(6SU]b!c!c!cdd e eer"cR|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 axisr?.).N).rr)r6rr7rqrItuplerGrHr8broadcast_arraysr;moveaxisr r<) rBrCrArIx1_shapex2_shapex1_x2_ress r#vecdotrs  x&&"(/*I*ICDDD rw DTBG^$uRX6HTBG^$uRX6H~$''QRRR"29bi88HC +c4 $ $C +c4 $ $C c4l c)n ,C :c)n % %%r"rd%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) rcz.Only floating-point dtypes are allowed in normNrc3$K|] }|v|V dS)Nr!).0inormalized_axiss r# zvector_norm..s-JJ1/1I1IQ1I1I1I1IJJr"c*g|]}j|Sr!rGrrrs r# zvector_norm..s///Qagaj///r")r6c*g|]}j|Sr!rrs r#rzvector_norm..s>X>X>Xaqwqz>X>X>Xr")rAr`r)r6rr7r;ravelrtrrrIranger8 transposer prodrDr r<r9relistrG) r2rAr_r`_axisrestnewshaperrGrrrs @@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^;; < rs""""""-,,,,,(((((( ////// JIIIIIIIIIIIIIIII     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 H00008 4 4 4 4/3 ] ] ] ] ] ]"AAAAcdEEEEEE)*Dffffff*46&&&&&&,OSejPQ++++++Z F F Fr"