U
|
O ±dG ã @ sð d dl mZ ddlmZmZ ddlmZ ddlmZ ddl m
|
Z
|
d dl m Z e rldd l mZmZmZmZmZ d d
|
l mZ d d lZd d lZG d d d eZG dd deZG dd deZG dd deZddddddddZdd ddd dd d d Z ddd!d"d#Z d d$dd dd%d&d'Z dd d!d(d)Z ddd!d*d+Z ddd!d,d-Z!dddd.d/d0Z"dd1d2ddd3dd4d5d6Z#dd dd7d8d9Z$d d:dd;dd<d=d>Z%ddd!d?d@Z&dddd.dAdBZ'd d:dd;dd<dCdDZ(dEdFddGddHdIdJZ)ddd!dKdLZ*dMdN Z+dddd.dOdPZ,dQdRddddSdTdUZ-ddVd!dWdXZ.ddYdddZdd[d\d]Z/d d$dd dd%d^d_Z0dd ddd dd d`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)ÚLiteralÚOptionalÚSequenceÚTupleÚUnion)Ú
|
NamedTupleNc @ s e Zd ZU ded< ded< dS )Ú
|
EighResultr Z eigenvaluesZ eigenvectorsN©Ú__name__Ú
|
__module__Ú __qualname__Ú__annotations__© r r úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/array_api/linalg.pyr s
|
r c @ s e Zd ZU ded< ded< dS )ÚQRResultr ÚQÚRNr r r r r r s
|
r c @ s e Zd ZU ded< ded< dS )Ú SlogdetResultr ÚsignZ logabsdetNr r r r r r s
|
r c @ s& e Zd ZU ded< ded< ded< dS )Ú SVDResultr ÚUÚSZVhNr r r r r r s
|
r F)Úupperr Úbool)Úxr! Úreturnc C s: | j tkrtdtj | j¡}|r0t |¡j S t |¡S )z
|
Array API compatible wrapper for :py:func:`np.linalg.cholesky <numpy.linalg.cholesky>`.
|
|
See its docstring for more information.
|
z2Only floating-point dtypes are allowed in cholesky)
|
Údtyper Ú TypeErrorÚnpÚlinalgÚcholeskyÚ_arrayr Ú_newZmT)r# r! ÚLr r r r) % s
|
r) éÿÿÿÿ©ÚaxisÚint)Úx1Úx2r/ r$ c C sr | j tks|j tkr td| j|jkr0td| jdkrBtd| j| dkrXtdt tj | j
|
|j
|
|d¡S )zz
|
Array API compatible wrapper for :py:func:`np.cross <numpy.cross>`.
|
|
See its docstring for more information.
|
z(Only numeric dtypes are allowed in crossz"x1 and x2 must have the same shaper z/cross() requires arrays of dimension at least 1é z cross() dimension must equal 3r. ) r% r r& ÚshapeÚ
|
ValueErrorÚndimr r+ r' Úcrossr* )r1 r2 r/ r r r r7 5 s
|
r7 )r# r$ c C s& | j tkrtdt tj | j¡¡S )z
|
Array API compatible wrapper for :py:func:`np.linalg.det <numpy.linalg.det>`.
|
|
See its docstring for more information.
|
z-Only floating-point dtypes are allowed in det) r% r r& r r+ r' r( Údetr* ©r# r r r r8 G s
|
r8 )Úoffset)r# r: r$ c C s t tj| j|ddd¡S )z
|
Array API compatible wrapper for :py:func:`np.diagonal <numpy.diagonal>`.
|
|
See its docstring for more information.
|
éþÿÿÿr- ©r: Zaxis1Zaxis2)r r+ r' Údiagonalr* ©r# r: r r r r= T s r= c C s, | j tkrtdtttjtj | j
|
¡ S )z
|
Array API compatible wrapper for :py:func:`np.linalg.eigh <numpy.linalg.eigh>`.
|
|
See its docstring for more information.
|
z.Only floating-point dtypes are allowed in eigh) r% r r& r Úmapr r+ r' r( Úeighr* r9 r r r r@ _ s
|
r@ c C s& | j tkrtdt tj | j¡¡S )z
|
Array API compatible wrapper for :py:func:`np.linalg.eigvalsh <numpy.linalg.eigvalsh>`.
|
|
See its docstring for more information.
|
z2Only floating-point dtypes are allowed in eigvalsh) r% r r& r r+ r' r( Úeigvalshr* r9 r r r rA o s
|
rA c C s& | j tkrtdt tj | j¡¡S )z
|
Array API compatible wrapper for :py:func:`np.linalg.inv <numpy.linalg.inv>`.
|
|
See its docstring for more information.
|
z-Only floating-point dtypes are allowed in inv) r% r r& r r+ r' r( Úinvr* r9 r r r rB | s
|
rB )r1 r2 r$ c C s2 | j tks|j tkr tdt t | j|j¡¡S )z|
|
Array API compatible wrapper for :py:func:`np.matmul <numpy.matmul>`.
|
|
See its docstring for more information.
|
z)Only numeric dtypes are allowed in matmul)r% r r& r r+ r' Úmatmulr* ©r1 r2 r r r rC s rC Zfro)ÚkeepdimsÚordz4Optional[Union[int, float, Literal[('fro', 'nuc')]]])r# rE rF r$ c C s. | j tkrtdt tjj| jd||d¡S )ú
|
Array API compatible wrapper for :py:func:`np.linalg.norm <numpy.linalg.norm>`.
|
|
See its docstring for more information.
|
z5Only floating-point dtypes are allowed in matrix_norm)r; r- ©r/ rE rF ) r% r r& r r+ r' r( Únormr* )r# rE rF r r r Ú matrix_norm s
|
rJ )r# Únr$ c C s( | j tkrtdt tj | j|¡¡S )z
|
Array API compatible wrapper for :py:func:`np.matrix_power <numpy.matrix_power>`.
|
|
See its docstring for more information.
|
zMOnly floating-point dtypes are allowed for the first argument of matrix_power) r% r r& r r+ r' r( Ú matrix_powerr* )r# rK r r r rL s
|
rL )Úrtolz Optional[Union[float, Array]])r# rM r$ c C sª | j dk rtj d¡tjj| jdd}|dkr`|jdddt| jd d
t |j ¡j
|
}n2t |t rp|j}|jdddt |¡d
|
tjf }t tj||kdd ¡S ) z
|
Array API compatible wrapper for :py:func:`np.matrix_rank <numpy.matrix_rank>`.
|
|
See its docstring for more information.
|
r zA1-dimensional array given. Array must be at least two-dimensionalF©Z
|
compute_uvNr- T)r/ rE r; .r. )r6 r' r( Z LinAlgErrorÚsvdr* Úmaxr4 Úfinfor% ÚepsÚ
|
isinstancer ÚasarrayZnewaxisr+ Z count_nonzero)r# rM r Ztolr r r Ú matrix_rank¼ s
|
0
|
"rU c C s( | j dk rtdt t | jdd¡¡S )Nr z5x must be at least 2-dimensional for matrix_transposer- r; )r6 r5 r r+ r' Zswapaxesr* r9 r r r Úmatrix_transposeÔ s
|
rV c C sN | j tks|j tkr td| jdks0|jdkr8tdt t | j |j ¡¡S )zz
|
Array API compatible wrapper for :py:func:`np.outer <numpy.outer>`.
|
|
See its docstring for more information.
|
z(Only numeric dtypes are allowed in outerr z/The input arrays to outer must be 1-dimensional)
|
r% r r& r6 r5 r r+ r' Úouterr* rD r r r rW Ú s
|
rW c C sR | j tkrtd|dkr:t| jdd
t | j ¡j }t tj
|
j | j |d¡S )z
|
Array API compatible wrapper for :py:func:`np.linalg.pinv <numpy.linalg.pinv>`.
|
|
See its docstring for more information.
|
z.Only floating-point dtypes are allowed in pinvNr; )Zrcond) r% r r& rP r4 r' rQ rR r r+ r( Úpinvr* )r# rM r r r rX ì s
|
|
rX Zreduced©Úmodez Literal[('reduced', 'complete')])r# rZ r$ c C s0 | j tkrtdtttjtjj | j
|
|d S )z
|
Array API compatible wrapper for :py:func:`np.linalg.qr <numpy.linalg.qr>`.
|
|
See its docstring for more information.
|
z,Only floating-point dtypes are allowed in qrrY ) r% r r& r r? r r+ r' r( Úqrr* )r# rZ r r r r[ ý s
|
r[ c C s, | j tkrtdtttjtj | j
|
¡ S )z
|
Array API compatible wrapper for :py:func:`np.linalg.slogdet <numpy.linalg.slogdet>`.
|
|
See its docstring for more information.
|
z1Only floating-point dtypes are allowed in slogdet) r% r r& r r? r r+ r' r( Úslogdetr* r9 r r r r\ s
|
r\ c C s¸ ddl m}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Ú isComplexTypeÚget_linalg_error_extobjÚ_raise_linalgerror_singular)Ú _umath_linalgr zDD->Dzdd->d)Ú signatureÚextobjF)Úcopy)Z linalg.linalgr] r^ r_ r` ra rb rc r( rd r6 Zsolve1ÚsolveZastype)ÚaÚbr] r^ r_ r` ra rb rc rd Ú_ÚwrapÚtZresult_tZgufuncre rf Úrr r r Ú_solve$ s $
|
ro c C s0 | j tks|j tkr tdt t| j|j¡S )z
|
Array API compatible wrapper for :py:func:`np.linalg.solve <numpy.linalg.solve>`.
|
|
See its docstring for more information.
|
z/Only floating-point dtypes are allowed in solve)r% r r& r r+ ro r* rD r r r rh ? s rh T©Ú full_matrices)r# rq r$ c C s0 | j tkrtdtttjtjj | j
|
|d S )z
|
Array API compatible wrapper for :py:func:`np.linalg.svd <numpy.linalg.svd>`.
|
|
See its docstring for more information.
|
z-Only floating-point dtypes are allowed in svdrp ) r% r r& r r? r r+ r' r( rO r* )r# rq r r r rO L s
|
rO z Union[Array, Tuple[Array, ...]]c C s* | j tkrtdt tjj| jdd¡S )Nz1Only floating-point dtypes are allowed in svdvalsFrN ) r% r r& r r+ r' r( rO r* r9 r r r Úsvdvals] s
|
rr ©Úaxesz/Union[int, Tuple[Sequence[int], Sequence[int]]])r1 r2 rt r$ c C s6 | j tks|j tkr tdt tj| j|j|d¡S )Nz,Only numeric dtypes are allowed in tensordotrs )r% r r& r r+ r' Ú tensordotr* )r1 r2 rt r r r ru e s ru c
|
C s2 | j tkrtdt t tj| j|ddd¡¡S )zz
|
Array API compatible wrapper for :py:func:`np.trace <numpy.trace>`.
|
|
See its docstring for more information.
|
z(Only numeric dtypes are allowed in tracer; r- r< ) r% r r& r r+ r' rT Útracer* r> r r r rv n s
|
rv c C sÊ | j tks|j tkr tdt| j|j}d|| j t| j }d||j t|j }|| || krrtdt | j
|
|j
|
¡\}}t ||d¡}t ||d¡}|dd d d
f |d }t |d ¡S )Nz)Only numeric dtypes are allowed in vecdot)r z6x1 and x2 must have the same size along the given axisr- .).N).r r )r% r r& rP r6 Útupler4 r5 r' Zbroadcast_arraysr* Zmoveaxisr r+ ) r1 r2 r/ r6 Zx1_shapeZx2_shapeZx1_Zx2_Úresr r r Úvecdot{ s ry rH z%Optional[Union[int, Tuple[int, ...]]]zOptional[Union[int, float]])r# r/ rE rF r$ c
|
s | j tkrtd| j |dkr. ¡ d}nt|tr¬t|| jtfddt jD }|| }t
|
|¡ t
|
j fdd|D tdf fdd|D ¡ d}n|}t t
|
jj ||d ¡}|rt| j}t|dkrìt | jn|| j}|D ] } d
|
|| < qút |t|}|S ) rG z.Only floating-point dtypes are allowed in normNr c 3 s | ]}| kr|V qd S )Nr ©Ú.0Úi)Únormalized_axisr r Ú <genexpr>© s z vector_norm.<locals>.<genexpr>c s g | ]} j | qS r )r4 rz )ri r r Ú
|
<listcomp>¬ s z vector_norm.<locals>.<listcomp>)r% )r/ rF r )r% r r& r* ZravelrS rw r r6 Úranger' Z transposer Úprodr0 r r+ r( rI Úlistr4 )
|
r# r/ rE rF Z_axisÚrestZnewshaperx r4 r| r )ri r} r Ú vector_norm s.
|
|
.ÿ
|
|
r )4Ú
|
__future__r Z_dtypesr r Z_manipulation_functionsr Z _array_objectr Z core.numericr Útypingr
|
Ú_typingr r r r r r Z numpy.linalgÚnumpyr' r r r r r) r7 r8 r= r@ rA rB rC rJ rL rU rV rW rX r[ r\ ro rh rO rr ru rv ry r Ú__all__r r r r Ú<module> sN -
|
