a XgG@sddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZdd lmZerdd lmZmZmZmZmZmZdd lmZdd lZdd lZGd ddeZGdddeZGdddeZ GdddeZ!ddddddddZ"dddddddd d!Z#ddd"d#d$Z$dd%dddd&d'd(Z%ddd"d)d*Z&ddd"d+d,Z'ddd"d-d.Z(dddd/d0d1Z)dd2d3ddd4dd5d6d7Z*dddd8d9d:Z+d d;ddd?Z,ddd"d@dAZ-dddd/dBdCZ.d d;dd`. See its docstring for more information. z2Only floating-point dtypes are allowed in cholesky) dtyper TypeErrornplinalgcholesky_arrayr _newZmTrr r )r)r'Lr%rrrr/-s  r/axisint)x1x2r5r*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 3r4) r+rr,shape ValueErrorndimr r1r-crossr0)r7r8r5rrrr=@s  r=)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,r r1r-r.detr0r)rrrr>Rs r>)offset)r)r@r*cCsttj|j|dddS)z Array API compatible wrapper for :py:func:`np.diagonal `. See its docstring for more information. r3)r@axis1axis2)r r1r-diagonalr0)r)r@rrrrD_srDcCs,|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,rmapr r1r-r.eighr0r?rrrrFjs rFcCs&|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,r r1r-r.eigvalshr0r?rrrrGzs rGcCs&|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,r r1r-r.invr0r?rrrrHs rH)r7r8r*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,r r1r-matmulr0r7r8rrrrIsrIZfro)keepdimsordz2Optional[Union[int, float, Literal['fro', 'nuc']]])r)rKrLr*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)rAr3r5rKrL) r+rr,r r1r-r.normr0)r)rKrLrrr matrix_norms rP)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,r r1r-r. matrix_powerr0)r)rQrrrrRs rR)rtolzOptional[Union[float, Array]])r)rSr*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. r zA1-dimensional array given. Array must be at least two-dimensionalFZ compute_uvNr3T)r5rKrA.r4)r<r-r.Z LinAlgErrorsvdr0maxr:finfor+eps isinstancer asarrayZnewaxisr1Z count_nonzero)r)rSr&Ztolrrr matrix_ranks  0 "r[cCs(|jdkrtdtt|jddS)Nr z5x must be at least 2-dimensional for matrix_transposer3rA)r<r;r r1r-Zswapaxesr0r?rrrmatrix_transposes r\cCsN|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,r<r;r r1r-outerr0rJrrrr]s r]cCsR|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 pinvNrA)Zrcond) r+rr,rVr:r-rWrXr r1r.pinvr0)r)rSrrrr^s   r^ZreducedmodezLiteral['reduced', 'complete'])r)r`r*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 qrr_) r+rr,rrEr r1r-r.qrr0)r)r`rrrras racCs,|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,r"rEr r1r-r.slogdetr0r?rrrrbs rbcCsddlm}m}m}m}m}m}m}ddlm } ||\}} ||||||\}} |||\} } |j dkrx| j }n| j }|| rdnd}t j|ddddd ||||d }Wdn1s0Y| |j| d d S) Nr ) _makearray_assert_stacked_2d_assert_stacked_square _commonType isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular) _umath_linalgrzDD->Dzdd->dcallignore)rkinvalidZoverdivideZunder) signatureF)copy)Z linalg.linalgrcrdrerfrgrhrir.rjr<Zsolve1solver-ZerrstateZastype)abrcrdrerfrgrhrirj_wraptZresult_tZgufuncrorrrr_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,r r1rxr0rJrrrrqKsrqT full_matrices)r)rzr*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 svdry) r+rr,r$rEr r1r-r.rUr0)r)rzrrrrUXs rUzUnion[Array, Tuple[Array, ...]]cCs*|jtvrtdttjj|jddS)Nz1Only floating-point dtypes are allowed in svdvalsFrT) r+rr,r r1r-r.rUr0r?rrrsvdvalsis r{axesz/Union[int, Tuple[Sequence[int], Sequence[int]]])r7r8r}r*cCs6|jtvs|jtvrtdttj|j|j|dS)Nz,Only numeric dtypes are allowed in tensordotr|)r+rr,r r1r- tensordotr0)r7r8r}rrrr~qsr~)r@r+zOptional[Dtype])r)r@r+r*c CsZ|jtvrtd|dur8|jtkr*t}n|jtkr8t}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 traceNrAr3)r@rBrCr+) r+rr,rrrr r r1r-rZtracer0)r)r@r+rrrrzs   rc Cs|jtvs|jtvrtdt|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 axisr3.).N).rr)r+rr,rVr<tupler:r;r-Zbroadcast_arraysr0Zmoveaxisr r1) r7r8r5r<Zx1_shapeZx2_shapeZx1_Zx2_resrrrvecdotsrrNz%Optional[Union[int, Tuple[int, ...]]]zOptional[Union[int, float]])r)r5rKrLr*c s|jtvrtd|j|dur.d}nt|trt||jtfddt jD}||}t | t j fdd|Dtdgfdd|DRd}n|}tt jj||d }|rt|j}t|durt |jn||j}|D] } d || <qt |t|}|S) rMz.Only floating-point dtypes are allowed in normNrc3s|]}|vr|VqdS)Nr.0i)normalized_axisrr zvector_norm..csg|]}j|qSr)r:r)rrrr rzvector_norm..)r+)r5rLr)r+rr,r0ZravelrYrrr<ranger-Z transposer prodr6r r1r.rOlistr:) r)r5rKrLZ_axisrestZnewshaperr:rr)rrrr vector_norms.    0  r)r/r=r>rDrFrGrHrIrPrRr[r\r]r^rarbrqrUr{r~rrr); __future__rZ_dtypesrrrrrr Z_manipulation_functionsr Z_elementwise_functionsr Z _array_objectr Z core.numericrtypingrZ_typingrrrrrrrZ numpy.linalgnumpyr-rrr"r$r/r=r>rDrFrGrHrIrPrRr[r\r]r^rarbrxrqrUr{r~rrr__all__rrrrsP               -