U mdE@sddlmZddlmZmZddlmZddlmZddl m Z ddl m Z e rldd l mZmZmZmZmZdd l mZdd lZdd lZGd d d eZGdddeZGdddeZGdddeZddddddddZddddddddd Zddd!d"d#Zdd$dddd%d&d'Zdd d!d(d)Zddd!d*d+Z ddd!d,d-Z!dddd.d/d0Z"dd1d2ddd3dd4d5d6Z#dddd7d8d9Z$d d:dd;dd<d=d>Z%ddd!d?d@Z&dddd.dAdBZ'd d:dd;dd<dCdDZ(dEdFddGddHdIdJZ)ddd!dKdLZ*dMdNZ+dddd.dOdPZ,dQdRddddSdTdUZ-ddVd!dWdXZ.ddYdddZdd[d\d]Z/dd$dddd%d^d_Z0dddddddd`daZ1d dddbddcdddddedfdgZ2dd d#d'd)d+d-d0d6d9d>d@dBdDdJdLdPdUdXd]d_dadggZ3d S)h) annotations)_floating_dtypes_numeric_dtypes)reshape)Array)normalize_axis_tuple) TYPE_CHECKING)LiteralOptionalSequenceTupleUnion) NamedTupleNc@seZdZUded<ded<dS) EighResultrZ eigenvaluesZ eigenvectorsN__name__ __module__ __qualname____annotations__rrO/home/sam/Atlas/atlas_env/lib/python3.8/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)xr!returncCs:|jtkrtdtj|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#r!Lrrrr)%s   r)axisint)x1x2r/r$cCsr|jtks|jtkrtd|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*)r1r2r/rrrr75s  r7)r#r$cCs&|jtkrtdttj|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#rrrr8Gs r8)offset)r#r:r$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=Tsr=cCs,|jtkrtdtttjtj |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*r9rrrr@_s r@cCs&|jtkrtdttj|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*r9rrrrAos rAcCs&|jtkrtdttj|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*r9rrrrB|s rB)r1r2r$cCs2|jtks|jtkrtdtt|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*r1r2rrrrCsrCZfro)keepdimsordz4Optional[Union[int, float, Literal[('fro', 'nuc')]]])r#rErFr$cCs.|jtkrtdttjj|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/rErF) r%rr&rr+r'r(normr*)r#rErFrrr matrix_norms rJ)r#nr$cCs(|jtkrtdttj|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#rKrrrrLs rL)rtolzOptional[Union[float, Array]])r#rMr$cCs|jdkrtjdtjj|jdd}|dkr`|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. rzA1-dimensional array given. Array must be at least two-dimensionalFZ compute_uvNr-T)r/rEr;.r.)r6r'r(Z LinAlgErrorsvdr*maxr4finfor%eps isinstancerasarrayZnewaxisr+Z count_nonzero)r#rMr Ztolrrr matrix_ranks  0 "rUcCs(|jdkrtdtt|jddS)Nrz5x must be at least 2-dimensional for matrix_transposer-r;)r6r5rr+r'Zswapaxesr*r9rrrmatrix_transposes rVcCsN|jtks|jtkrtd|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&r6r5rr+r'outerr*rDrrrrWs rWcCsR|jtkrtd|dkr: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 pinvNr;)Zrcond) r%rr&rPr4r'rQrRrr+r(pinvr*)r#rMrrrrXs   rXZreducedmodez Literal[('reduced', 'complete')])r#rZr$cCs0|jtkrtdtttjtjj |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 qrrY) r%rr&rr?rr+r'r(qrr*)r#rZrrrr[s r[cCs,|jtkrtdtttjtj |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*r9rrrr\ s r\cCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkrx| j }n| j }|| rdnd}||}|||||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.linalgr]r^r_r`rarbrcr(rdr6Zsolve1solveZastype)abr]r^r_r`rarbrcrd_wraptZresult_tZgufuncrerfrrrr_solve$s$    rocCs0|jtks|jtkrtdtt|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+ror*rDrrrrh?srhT full_matrices)r#rqr$cCs0|jtkrtdtttjtjj |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 svdrp) r%rr&rr?rr+r'r(rOr*)r#rqrrrrOLs rOzUnion[Array, Tuple[Array, ...]]cCs*|jtkrtdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrN) r%rr&rr+r'r(rOr*r9rrrsvdvals]s rraxesz/Union[int, Tuple[Sequence[int], Sequence[int]]])r1r2rtr$cCs6|jtks|jtkrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotrs)r%rr&rr+r' tensordotr*)r1r2rtrrrruesruc Cs2|jtkrtdtttj|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'rTtracer*r>rrrrvns rvc Cs|jtks|jtkrtdt|j|j}d||jt|j}d||jt|j}||||krrtdt |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&rPr6tupler4r5r'Zbroadcast_arraysr*Zmoveaxisrr+) r1r2r/r6Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdot{sryrHz%Optional[Union[int, Tuple[int, ...]]]zOptional[Union[int, float]])r#r/rErFr$c s|jtkrtd|j|dkr.d}nt|trt||jtfddt jD}||}t | t j fdd|Dtdffdd|Dd}n|}tt jj||d }|rt|j}t|dkrt |jn||j}|D] } d || <qt |t|}|S) rGz.Only floating-point dtypes are allowed in normNrc3s|]}|kr|VqdS)Nr.0i)normalized_axisrr szvector_norm..csg|]}j|qSr)r4rz)rirr szvector_norm..)r%)r/rFr)r%rr&r*ZravelrSrwr r6ranger'Z transposerprodr0rr+r(rIlistr4) r#r/rErFZ_axisrestZnewshaperxr4r|r)rir}r vector_norms.    .  r)4 __future__rZ_dtypesrrZ_manipulation_functionsrZ _array_objectrZ core.numericr typingr Z_typingr r r rrrZ numpy.linalgnumpyr'rrrrr)r7r8r=r@rArBrCrJrLrUrVrWrXr[r\rorhrOrrrurvryr__all__rrrrsN             -