o ʩZhG@s2ddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZerFdd lmZmZmZmZmZmZdd lmZdd lZdd lZGd ddeZGdddeZGdddeZ GdddeZ!dddoddZ"dddpd$d%Z#dqd&d'Z$dd(drd*d+Z%dsd,d-Z&dqd.d/Z'dqd0d1Z(dtd2d3Z)dd4d5dud9d:Z*dvddwdAdBZ,dqdCdDZ-dtdEdFZ.d d>dwdGdHZ/dIdJdxdMdNZ0dydOdPZ1dQdRZ2dtdSdTZ3dUdVdzdXdYZ4d{d[d\Z5dd]d|d`daZ6dd dbd}dedfZ7dddpdgdhZ8d dddid~dldmZ9gdnZ:d S)) annotations)_floating_dtypes_numeric_dtypesfloat32float64 complex64 complex128)reshape)conj)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnionDtype) NamedTupleNc@eZdZUded<ded<dS) EighResultr Z eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrM/var/www/html/lang_env/lib/python3.10/site-packages/numpy/array_api/linalg.pyr  rc@r)QRResultr QRNrrrrrr!r r!c@r) SlogdetResultr signZ logabsdetNrrrrrr$"r r$c@s&eZdZUded<ded<ded<dS) SVDResultr USZVhNrrrrrr&&s  r&F)upperxr r)boolreturncCsT|jtvr tdtj|j}|r%t|j }|jt t fvr#t |}|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 _newZmTrr r )r*r)Lr'rrrr1-s   r1axisx1x2r7intcCsr|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 3r6) r-rr.shape ValueErrorndimr r3r/crossr2)r8r9r7rrrr?@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 r3r/r0detr2r*rrrrARs rA)offsetrCcCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r5)rCaxis1axis2)r r3r/diagonalr2)r*rCrrrrG_srGcC,|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 r3r/r0eighr2rBrrrrJj rJcCr@)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 r3r/r0eigvalshr2rBrrrrLz rLcCr@)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 r3r/r0invr2rBrrrrNrMrNcCs2|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 r3r/matmulr2r8r9rrrrOsrOZfro)keepdimsordrQrR2Optional[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)rDr5r7rQrR) r-rr.r r3r/r0normr2)r*rQrRrrr matrix_norms rWncCs(|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 r3r/r0 matrix_powerr2)r*rXrrrrYs rY)rtolrZOptional[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_uvNr5T)r7rQrD.r6)r>r/r0Z LinAlgErrorsvdr2maxr<finfor-eps isinstancer asarrayZnewaxisr3Z count_nonzero)r*rZr(Ztolrrr matrix_ranks  0 "rccCs(|jdkr tdtt|jddS)Nr z5x must be at least 2-dimensional for matrix_transposer5rD)r>r=r r3r/Zswapaxesr2rBrrrmatrix_transposes rdcCsN|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 r3r/outerr2rPrrrres recCsR|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 pinvNrD)Zrcond) r-rr.r^r<r/r_r`r r3r0pinvr2)r*rZrrrrfs  rfZreducedmoderhLiteral['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 qrrg) r-rr.r!rIr r3r/r0qrr2)r*rhrrrrk rkcCrH)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$rIr r3r/r0slogdetr2rBrrrrmrKrmcCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkr<| j }n| j }|| rEdnd}t j|ddddd ||||d }Wdn1scwY| |j| d d S) Nr ) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->dcallignore)rvinvalidZoverdivideZunder) signatureF)copy)Z linalg.linalgrnrorprqrrrsrtr0rur>Zsolve1solver/ZerrstateZastype)abrnrorprqrrrsrtru_wraptZresult_tZgufuncrzrrrr_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 r3rr2rPrrrr|Ksr|T full_matricesrcCrj)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&rIr r3r/r0r]r2)r*rrrrr]Xrlr]Union[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFr\) r-rr.r r3r/r0r]r2rBrrrsvdvalsis 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 r3r/ tensordotr2)r8r9rrrrrqsr)rCr-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 traceNrDr5)rCrErFr-) r-rr.rrrr r r3r/rbtracer2)r*rCr-rrrrzs   "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 axisr5.).N).rr)r-rr.r^r>tupler<r=r/Zbroadcast_arraysr2Zmoveaxisr r3) r8r9r7r>Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdotsrrU%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) rTz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr.0i)normalized_axisrr szvector_norm..csg|]}j|qSr)r<r)r}rr szvector_norm..)r-)r7rRr)r-rr.r2Zravelrarrr>ranger/Z transposer prodr:r r3r0rVlistr<) r*r7rQrRZ_axisrestZnewshaperr<rr)r}rr vector_norms.    0  r)r1r?rArGrJrLrNrOrWrYrcrdrerfrkrmr|r]rrrrr)r*r r)r+r,r )r8r r9r r7r:r,r )r*r r,r )r*r rCr:r,r )r*r r,r)r8r r9r r,r )r*r rQr+rRrSr,r )r*r rXr:r,r )r*r rZr[r,r )r*r rhrir,r!)r*r r,r$)r*r rr+r,r&)r*r r,r)r8r r9r rrr,r )r*r rCr:r-rr,r ) r*r r7rrQr+rRrr,r ); __future__rZ_dtypesrrrrrr Z_manipulation_functionsr Z_elementwise_functionsr Z _array_objectr Z core.numericrtypingrZ_typingrrrrrrrZ numpy.linalgnumpyr/rr!r$r&r1r?rArGrJrLrNrOrWrYrcrdrerfrkrmrr|r]rrrrr__all__rrrrsR                   -