o $eC@sddlmZddlmZmZddlmZddlmZer*ddl m Z m Z m Z m Z mZddlmZddlZddlZGd d d eZGd d d eZGd ddeZGdddeZdddiddZdddjd d!Zdkd"d#Zdd$dld&d'Zdmd(d)Zdkd*d+Zdkd,d-Zdnd.d/Zdd0d1dod5d6Zdpd8d9Z dd:dqd=d>Z!dkd?d@Z"dndAdBZ#dd:dqdCdDZ$dEdFdrdIdJZ%dsdKdLZ&dMdNZ'dndOdPZ(dQdRdtdTdUZ)dudWdXZ*dYdZdvd]d^Z+dd$dld_d`Z,dddjdadbZ-dddYdcdwdfdgZ.gdhZ/dS)x) annotations)_floating_dtypes_numeric_dtypes)Array) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc@eZdZUded<ded<dS) EighResultrZ eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrFD:\Projects\ConvertPro\env\Lib\site-packages\numpy/array_api/linalg.pyr  rc@r)QRResultrQRNrrrrrrrrc@r) SlogdetResultrsignZ logabsdetNrrrrrrrrc@s&eZdZUded<ded<ded<dS) SVDResultrUSZVhNrrrrrrs  rF)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(axisx1x2r.intcCsr|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 3r-) r$rr%shape ValueErrorndimrr*r&crossr))r/r0r.rrrr62s  r6cC&|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!rrrr8Ds r8)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=Qsr=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)r9rrrrA\ rAcCr7)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)r9rrrrCl rCcCr7)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)r9rrrrEyrDrEcCs2|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)r/r0rrrrFsrFZfro)keepdimsordrHrI2Optional[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,r.rHrI) r$rr%rr*r&r'normr))r!rHrIrrr matrix_norms rNncCs(|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!rOrrrrPs rP)rtolrQOptional[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. zA1-dimensional array given. Array must be at least two-dimensionalFZ compute_uvNr,T)r.rHr;.r-)r5r&r'Z LinAlgErrorsvdr)maxr3finfor$eps isinstancerasarrayZnewaxisr*Z count_nonzero)r!rQrZtolrrr matrix_ranks  0 "r[cCs(|jdkr tdtt|jddS)NrSz5x must be at least 2-dimensional for matrix_transposer,r;)r5r4rr*r&Zswapaxesr)r9rrrmatrix_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%r5r4rr*r&outerr)rGrrrr]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%rVr3r&rWrXrr*r'pinvr))r!rQrrrr^s  r^Zreducedmoder`Literal['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 qrr_) r$rr%rr@rr*r&r'qrr))r!r`rrrrc rccCr?)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%rr@rr*r&r'slogdetr)r9rrrre rBrecCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkr<| j }n| j }|| rEdnd}||}|||||d}| |j | dd S) NrS) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->d) signatureextobjF)copy)Z linalg.linalgrfrgrhrirjrkrlr'rmr5Zsolve1solveZastype)abrfrgrhrirjrkrlrm_wraptZresult_tZgufuncrnrorrrr_solve!s$    rxcCs0|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*rxr)rGrrrrq<srqT full_matricesrzcCrb)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 svdry) r$rr%rr@rr*r&r'rUr))r!rzrrrrUIrdrUUnion[Array, Tuple[Array, ...]]cCs*|jtvr tdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrT) r$rr%rr*r&r'rUr)r9rrrsvdvalsZs r|rSaxesr~/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))r/r0r~rrrrbsrc 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&rZtracer)r>rrrrks  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%rVr5tupler3r4r&Zbroadcast_arraysr)Zmoveaxisrr*) r/r0r.r5Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdotxsrrL%Optional[Union[int, Tuple[int, ...]]]Optional[Union[int, float]]cs|jtvr td|jdurdn7ttrNtfddtjD}|}t | t fddDgfdd|DRdt t jj||dS) rKz.Only floating-point dtypes are allowed in normNrc3s|] }|vr|VqdS)Nr.0ir-rr szvector_norm..csg|]}j|qSr)r3r)rrrr szvector_norm..rL)r$rr%r)flattenrYrranger5r&Z transposeZreshapeprodrr*r'rM)r!r.rHrIrestZnewshaper)rrr.r vector_norms  <r)r(r6r8r=rArCrErFrNrPr[r\r]r^rcrerqrUr|rrrr)r!rr r"r#r)r/rr0rr.r1r#r)r!rr#r)r!rr:r1r#r)r!rr#r)r/rr0rr#r)r!rrHr"rIrJr#r)r!rrOr1r#r)r!rrQrRr#r)r!rr`rar#r)r!rr#r)r!rrzr"r#r)r!rr#r{)r/rr0rr~rr#r) r!rr.rrHr"rIrr#r)0 __future__rZ_dtypesrrZ _array_objectrtypingrZ_typingrr r r r r Z numpy.linalgnumpyr&rrrrr(r6r8r=rArCrErFrNrPr[r\r]r^rcrerxrqrUr|rrrr__all__rrrrsL