o Je-G@s&ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZer@dd lmZmZmZmZmZmZdd lmZdd lZdd lZGd d d eZGdddeZGdddeZGdddeZdddnddZ dddod#d$Z!dpd%d&Z"dd'dqd)d*Z#drd+d,Z$dpd-d.Z%dpd/d0Z&dsd1d2Z'dd3d4dtd8d9Z(dud;d<Z)d d=dvd@dAZ*dpdBdCZ+dsdDdEZ,d d=dvdFdGZ-dHdIdwdLdMZ.dxdNdOZ/dPdQZ0dsdRdSZ1dTdUdydWdXZ2dzdZd[Z3dd\d{d_d`Z4dd dad|dddeZ5dddodfdgZ6d dddhd}dkdlZ7gdmZ8d S)~) annotations)_floating_dtypes_numeric_dtypesfloat32float64 complex64 complex128)reshape)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnionDtype) NamedTupleNc@eZdZUded<ded<dS) EighResultr Z eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrd/home/seandotau/au.memorygallery.club/gallery/lib/python3.10/site-packages/numpy/array_api/linalg.pyr  rc@r)QRResultr QRNrrrrrr rr c@r) SlogdetResultr signZ logabsdetNrrrrrr#!rr#c@s&eZdZUded<ded<ded<dS) SVDResultr USZVhNrrrrrr%%s  r%F)upperxr r(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(Lrrrr0,s   r0axisx1x2r6intcCsr|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 ValueErrorndimr r2r.crossr1)r7r8r6rrrr><s  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-r r2r.r/detr1r)rrrr@Ns r@)offsetrBcCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r4)rBaxis1axis2)r r2r.diagonalr1)r)rBrrrrF[srFcC,|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-rmapr r2r.r/eighr1rArrrrIf rIcCr?)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-r r2r.r/eigvalshr1rArrrrKv rKcCr?)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-r r2r.r/invr1rArrrrMrLrMcCs2|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-r r2r.matmulr1r7r8rrrrNsrNZfro)keepdimsordrPrQ2Optional[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)rCr4r6rPrQ) r,rr-r r2r.r/normr1)r)rPrQrrr matrix_norms rVncCs(|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-r r2r.r/ matrix_powerr1)r)rWrrrrXs rX)rtolrYOptional[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-dimensionalFZ compute_uvNr4T)r6rPrC.r5)r=r.r/Z LinAlgErrorsvdr1maxr;finfor,eps isinstancer asarrayZnewaxisr2Z count_nonzero)r)rYr'Ztolrrr matrix_ranks  0 "rbcCs(|jdkr tdtt|jddS)Nr z5x must be at least 2-dimensional for matrix_transposer4rC)r=r<r r2r.Zswapaxesr1rArrrmatrix_transposes rccCsN|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<r r2r.outerr1rOrrrrds rdcCsR|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)Zrcond) r,rr-r]r;r.r^r_r r2r/pinvr1)r)rYrrrres  reZreducedmodergLiteral['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 qrrf) r,rr-r rHr r2r.r/qrr1)r)rgrrrrj rjcCrG)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#rHr r2r.r/slogdetr1rArrrrlrJrlcCsddlm}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.linalgrmrnrorprqrrrsr/rtr=Zsolve1solveZastype)abrmrnrorprqrrrsrt_wraptZresult_tZgufuncrurvrrrr_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-r r2rr1rOrrrrxFsrxT full_matricesrcCri)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%rHr r2r.r/r\r1)r)rrrrr\Srkr\Union[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFr[) r,rr-r r2r.r/r\r1rArrrsvdvalsds raxesr/Union[int, Tuple[Sequence[int], Sequence[int]]]cCs6|jtvs |jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr)r,rr-r r2r. tensordotr1)r7r8rrrrrlsr)rBr,r,Optional[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 traceNrCr4)rBrDrEr,) r,rr-rrrr r r2r.ratracer1)r)rBr,rrrrus   "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-r]r=tupler;r<r.Zbroadcast_arraysr1Zmoveaxisr r2) r7r8r6r=Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdotsrrT%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) rSz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr.0i)normalized_axisrr szvector_norm..csg|]}j|qSr)r;r)ryrr szvector_norm..)r,)r6rQr)r,rr-r1Zravelr`rr r=ranger.Z transposer prodr9r r2r/rUlistr;) r)r6rPrQZ_axisrestZnewshaperr;rr)ryrr vector_norms.    0  r)r0r>r@rFrIrKrMrNrVrXrbrcrdrerjrlrxr\rrrrr)r)r r(r*r+r )r7r r8r r6r9r+r )r)r r+r )r)r rBr9r+r )r)r r+r)r7r r8r r+r )r)r rPr*rQrRr+r )r)r rWr9r+r )r)r rYrZr+r )r)r rgrhr+r )r)r r+r#)r)r rr*r+r%)r)r r+r)r7r r8r rrr+r )r)r rBr9r,rr+r ) r)r r6rrPr*rQrr+r )9 __future__rZ_dtypesrrrrrr Z_manipulation_functionsr Z _array_objectr Z core.numericr typingrZ_typingrrrrrrrZ numpy.linalgnumpyr.rr r#r%r0r>r@rFrIrKrMrNrVrXrbrcrdrerjrlrrxr\rrrrr__all__rrrrsP                  -