a hEb?@sddlmZddlmZmZddlmZddlmZerTddl m Z m Z m Z m Z mZddlmZddlZddlZGd d d eZGd d d eZGd ddeZGdddeZddddddddZdddddddddZddddd Zdd!dddd"d#d$Zdd dd%d&Zdddd'd(Zdddd)d*Zdddd+d,d-Zdd.d/ddd0dd1d2d3Zdddd4d5d6Z dd7dd8dd9d:d;Z!ddddd?Z#dd7dd8dd9d@dAZ$dBdCddDd dEdFdGZ%ddddHdIZ&dJdKZ'dddd+dLdMZ(dNdOddddPdQdRZ)ddSddTdUZ*dVdWdddXddYdZd[Z+dd!dddd"d\d]Z,dddddddd^d_Z-dddVd`ddaddbddcdddeZ.gdfZ/dS)g) annotations)_floating_dtypes_numeric_dtypes)Array) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc@seZdZUded<ded<dS) EighResultrZ eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrV/home/ja/django-apps/lartica_env/lib/python3.9/site-packages/numpy/array_api/linalg.pyrs rc@seZdZUded<ded<dS)QRResultrQRNrrrrrrs rc@seZdZUded<ded<dS) SlogdetResultrsignZ logabsdetNrrrrrrs rc@s&eZdZUded<ded<ded<dS) SVDResultrUSZVhNrrrrrrs rF)upperrbool)xrreturncCs:|jtvrtdtj|j}|r0t|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 rLrrrr&"s   r&axisint)x1x2r,r!cCsr|jtvs|jtvrtd|j|jkr0td|jdkrBtd|j|dkrXtdttj |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.r/r,rrrr42s  r4)r r!cCs&|jtvrtdttj|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 rrrr6Ds r6)offset)r r8r!cCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r*r8Zaxis1Zaxis2)rr(r$diagonalr'r r8rrrr;Qsr;cCs,|jtvrtdtttjtj |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'r7rrrr>]s r>cCs&|jtvrtdttj|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) r"rr#rr(r$r%eigvalshr'r7rrrr?ns r?cCs&|jtvrtdttj|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) r"rr#rr(r$r%invr'r7rrrr@{s r@)r.r/r!cCs2|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.r/rrrrAsrAZfro)keepdimsordz2Optional[Union[int, float, Literal['fro', 'nuc']]])r rCrDr!cCs.|jtvrtdttjj|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)r9r*r,rCrD) r"rr#rr(r$r%normr')r rCrDrrr matrix_norms rH)r nr!cCs(|jtvrtdttj|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 rIrrrrJs rJ)rtolzOptional[Union[float, Array]])r rKr!cCs|jdkrtjdtjj|jdd}|dur`|jdddt|jd dt|j j }n2t |t rp|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,rCr9.r+)r3r$r%Z LinAlgErrorsvdr'maxr1finfor"eps isinstancerasarrayZnewaxisr(Z count_nonzero)r rKrZtolrrr matrix_ranks  0 "rTcCs(|jdkrtdtt|jddS)NrLz5x must be at least 2-dimensional for matrix_transposer*r9)r3r2rr(r$Zswapaxesr'r7rrrmatrix_transposes rUcCsN|jtvs|jtvrtd|jdks0|jdkr8tdtt|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#r3r2rr(r$outerr'rBrrrrVs rVcCsR|jtvrtd|dur:t|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 pinvNr9)Zrcond) r"rr#rOr1r$rPrQrr(r%pinvr')r rKrrrrWs   rWZreducedmodezLiteral['reduced', 'complete'])r rYr!cCs0|jtvrtdtttjtjj |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 qrrX) r"rr#rr=rr(r$r%qrr')r rYrrrrZs rZcCs,|jtvrtdtttjtj |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) r"rr#rr=rr(r$r%slogdetr'r7rrrr[ s r[cCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkrx| j }n| j }|| rdnd}||}|||||d}| |j | dd S) NrL) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->d) signatureextobjF)copy)Z linalg.linalgr\r]r^r_r`rarbr%rcr3Zsolve1solveZastype)abr\r]r^r_r`rarbrc_wraptZresult_tZgufuncrdrerrrr_solve#s$    rncCs0|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(rnr'rBrrrrg>srgT full_matrices)r rpr!cCs0|jtvrtdtttjtjj |j |dS)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 svdro) r"rr#rr=rr(r$r%rNr')r rprrrrNKs rNzUnion[Array, Tuple[Array, ...]]cCsttjj|jddS)NFrM)rr(r$r%rNr'r7rrrsvdvals\srqrLaxesz/Union[int, Tuple[Sequence[int], Sequence[int]]])r.r/rsr!cCs6|jtvs|jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotrr)r"rr#rr(r$ tensordotr')r.r/rsrrrrtbsrtc Cs tttj|j|dddS)zz Array API compatible wrapper for :py:func:`np.trace `. See its docstring for more information. r9r*r:)rr(r$rStracer'r<rrrruksrucCst|||f|ffdS)Nrr)rtr5rrrvecdotvsrvrFz%Optional[Union[int, Tuple[int, int]]]zOptional[Union[int, float]])r r,rCrDr!cs|jtvrtd|jdur.dnnttrtfddtjD}|}t | t fddDgfdd|DRdt t jj||dS) rEz.Only floating-point dtypes are allowed in normNrc3s|]}|vr|VqdS)Nr.0ir+rr zvector_norm..csg|]}j|qSr)r1rw)rhrr r{zvector_norm..rF)r"rr#r'flattenrRtupleranger3r$Z transposeZreshapeprodrr(r%rG)r r,rCrDrestZnewshaper)rhr,r vector_norms  <r)r&r4r6r;r>r?r@rArHrJrTrUrVrWrZr[rgrNrqrtrurvr)0 __future__rZ_dtypesrrZ _array_objectrtypingrZ_typingrr r r r r Z numpy.linalgnumpyr$rrrrr&r4r6r;r>r?r@rArHrJrTrUrVrWrZr[rnrgrNrqrtrurvr__all__rrrrsJ