= ( d dl mZ ddlmZmZmZmZmZmZ ddl m
Z
ddl m Z ddl
mZ d dlmZ erdd lmZmZmZmZmZmZ d d
lmZ d d lZd d lZ G d d
e Z G d de Z G d de Z G d de Z dddVdZ d d dWd"Z!dXd#Z"d d$dYd&Z#dZd'Z$dXd(Z%dXd)Z&d[d*Z'dd+d,d\d0Z(d]d2Z)d d3d^d6Z*dXd7Z+d[d8Z,d d3d^d9Z-d:d;d_d>Z.d`d?Z/d@ Z0d[dAZ1dBdCdadEZ2dbdGZ3ddHdcdKZ4d d dLdddOZ5d d dWdPZ6d dddQdedTZ7g dUZ8d S )f ) annotations )_floating_dtypes_numeric_dtypesfloat32float64 complex64
complex128)reshape)Array )normalize_axis_tuple)
TYPE_CHECKING)LiteralOptionalSequenceTupleUnionDtype)
NamedTupleNc $ e Zd ZU ded< ded< dS )
EighResultr eigenvalues eigenvectorsN__name__
__module__ __qualname____annotations__ j/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/linalg.pyr r s* r! r c $ e Zd ZU ded< ded< dS )QRResultr QRNr r r! r" r$ r$ s" HHH HHHHHr! r$ c $ e Zd ZU ded< ded< dS )
SlogdetResultr sign logabsdetNr r r! r" r( r( ! s' KKKr! r( c . e Zd ZU ded< ded< ded< dS ) SVDResultr USVhNr r r! r" r, r, % s+ HHH HHH
IIIIIr! r, F)upperxr r0 boolreturnc | j t vrt d t j | j }|rt j | j S t j | 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 _newmT)r1 r0 Ls r" r9 r9 , s_ w&&&LMMM
18$$A z!}} :a==r! axisx1x2r@ intc p | j t vs|j t vrt d | j |j k rt d | j dk rt d | j | dk rt d t
j t j | j
|j
| S )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 shaper z/cross() requires arrays of dimension at least 1 z cross() dimension must equal 3r? ) r5 r r6 shape
ValueErrorndimr r; r7 crossr: )rA rB r@ s r" rI rI < s
x&&"(/*I*IBCCC x28=>>> w!||JKKK x~9::: :bhry")$??? @ @@r! c | j t vrt d t j t
j | j S )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) r5 r r6 r r; r7 r8 detr: r1 s r" rK rK N sA w&&&GHHH :bimmAH-- . ..r! )offsetrM c ` t j t j | j |dd S )z
Array API compatible wrapper for :py:func:`np.diagonal `.
See its docstring for more information.
r> )rM axis1axis2)r r; r7 diagonalr: )r1 rM s r" rR rR [ s) :bk!(62NNN O OOr! c | j t vrt d t t t
j t j | 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) r5 r r6 r mapr r; r7 r8 eighr: rL s r" rU rU f sJ w&&&HIII s5:ry~~ah'?'?@@ AAr! c | j t vrt d t j t
j | j S )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) r5 r r6 r r; r7 r8 eigvalshr: rL s r" rW rW v sC w&&&LMMM :bi((22 3 33r! c | j t vrt d t j t
j | j S )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) r5 r r6 r r; r7 r8 invr: rL s r" rY rY sA w&&&GHHH :bimmAH-- . ..r! c | j t vs|j t vrt d t j t j | j |j S )z|
Array API compatible wrapper for :py:func:`np.matmul `.
See its docstring for more information.
z)Only numeric dtypes are allowed in matmul)r5 r r6 r r; r7 matmulr: rA rB s r" r[ r[ sL
x&&"(/*I*ICDDD :bi 2955 6 66r! fro)keepdimsordr^ r_ 2Optional[Union[int, float, Literal['fro', 'nuc']]]c | j t vrt d t j t
j | j d|| S )
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)rO r> r@ r^ r_ ) r5 r r6 r r; r7 r8 normr: )r1 r^ r_ s r" matrix_normre sJ w&&&OPPP :binnQXHxUXnYY Z ZZr! nc | j t vrt d t j t
j | 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) r5 r r6 r r; r7 r8 matrix_powerr: )r1 rf s r" rh rh sG w&&&ghhh :bi,,QXq99 : ::r! )rtolri Optional[Union[float, Array]]c Z | j dk r t j d t j | j d }|T| dd t
| j d d z t j |j j
z }nZt |t r|j }| dd t j
| d
t j f z }t j t j ||k d 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-dimensionalF
compute_uvNr> T)r@ r^ rO .r? )rH r7 r8 LinAlgErrorsvdr: maxrF finfor5 eps
isinstancer asarraynewaxisr;
count_nonzero)r1 ri r. tols r" matrix_rankrx s vzzi##$ghhh
ah5
11A |eede++c!'"##,.?.??"(17BSBSBWW dE " " ;D eede++BJt,<, rO )rH rG r r; r7 swapaxesr: rL s r" matrix_transposer{ s< vzzPQQQ :bk!(B33 4 44r! c | j t vs|j t vrt d | j dk s |j dk rt d t j t j | 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 outerr z/The input arrays to outer must be 1-dimensional)
r5 r r6 rH rG r r; r7 outerr: r\ s r" r} r} sr
x&&"(/*I*IBCCC
w!||rw!||JKKK :bhry")44 5 55r! c & | j t vrt d |;t | j dd t j | j j z }t j t
j
| j | S )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 pinvNrO )rcond)
r5 r r6 rp rF r7 rq rr r r; r8 pinvr: )r1 ri s r" r r sw w&&&HIII |17233< 28AG#4#4#88 :binnQXTn:: ; ;;r! reducedmoder Literal['reduced', 'complete']c | j t vrt d t t t
j t j | j
| S )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 ) r5 r r6 r$ rT r r; r7 r8 qrr: )r1 r s r" r r sN w&&&FGGG SRY\\!(\%F%FGG HHr! c | j t vrt d t t t
j t j | 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) r5 r r6 r( rT r r; r7 r8 slogdetr: rL s r" r r sL w&&&KLLL #ej")*;*;AH*E*EFF GGr! c z ddl m}m}m}m}m}m}m} ddlm } || \ } }
|| || || \ }} || | \ } }
|j
dk r| j }n| j } || rdnd} || } || ||| } | |
|
d S )
Nr
)
_makearray_assert_stacked_2d_assert_stacked_square _commonType
isComplexTypeget_linalg_error_extobj_raise_linalgerror_singular)
_umath_linalgr zDD->Dzdd->d) signatureextobjF)copy)
linalg.linalgr r r r r r r r8 r rH solve1solveastype)abr r r r r r r r _wraptresult_tgufuncr r rs r" _solver + sK > > > > > > > > > > > > > > > > > > '&&&&& :a==DAqq1 jmmGAt +a##KAx v{{ % $ )=++8I
$
$%@
A
AFq!y888A 4.. / //r! c | j t vs|j t vrt d t j t | j |j S )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)r5 r r6 r r; r r: r\ s r" r r F sL
x'''28;K+K+KIJJJ :fRY 22 3 33r! T
full_matricesr c | j t vrt d t t t
j t j | j
| S )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 svdr ) r5 r r6 r, rT r r; r7 r8 ro r: )r1 r s r" ro ro S sN w&&&GHHH c%*bimmAHMm&Z&Z[[ \\r! Union[Array, Tuple[Array, ...]]c | j t vrt d t j t
j | j d S )Nz1Only floating-point dtypes are allowed in svdvalsFrl ) r5 r r6 r r; r7 r8 ro r: rL s r" svdvalsr d sC w&&&KLLL :bimmAHm?? @ @@r! axesr /Union[int, Tuple[Sequence[int], Sequence[int]]]c | j t vs|j t vrt d t j t j | j |j | S )Nz,Only numeric dtypes are allowed in tensordotr )r5 r r6 r r; r7 tensordotr: )rA rB r s r" r r l sQ
x&&"(/*I*IFGGG :bl29bidCCC D DDr! )rM r5 r5 Optional[Dtype]c " | j t vrt d |/| j t k rt }n| j t
k rt }t j t j
t j | j |dd| S )zz
Array API compatible wrapper for :py:func:`np.trace `.
See its docstring for more information.
z(Only numeric dtypes are allowed in traceNrO r> )rM rP rQ r5 )
r5 r r6 r r r r
r r; r7 rt tracer: )r1 rM r5 s r" r r u s wo%%BCCC
} 7g EE
W
!
! E :bj!(6SU]b!c!c!cdd e eer! c R | j t vs|j t vrt d t | j |j }d|| j z
z t | j z }d||j z
z t |j z }|| || k rt d t j | j
|j
\ }}t j ||d }t j ||d }|dd d d f |d z }t j
|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 )r5 r r6 rp rH tuplerF rG r7 broadcast_arraysr: moveaxisr r; ) rA rB r@ rH x1_shapex2_shapex1_x2_ress r" vecdotr s
x&&"(/*I*ICDDD rw DTBG^$uRX6HTBG^$uRX6H~$ ''QRRR"29bi88HC
+c4
$
$C
+c4
$
$C
c4l
c)n
,C :c)n % %%r! rc %Optional[Union[int, Tuple[int, ...]]]Optional[Union[int, float]]c 8
| j t vrt d | j
|
d}nt |t rt || j t
fdt
j D }||z }t j
| t j
fd|D t g
fd|D R
d}n|}t j t j
|| }|rft' | j }t |t | j n|| j }|D ]} d || < t |t
| }|S )
rb z.Only floating-point dtypes are allowed in normNr c 3 $ K | ]
}|v|V d S )Nr ).0inormalized_axiss r" z vector_norm.. s- JJ1/1I1IQ1I1I1I1IJJr! c * g | ]}j | S r rF r r r s r"
z vector_norm.. s ///Qagaj///r! )r5 c * g | ]}j | S r r r s r" r z vector_norm.. s >X>X>Xaqwqz>X>X>Xr! )r@ r_ r )r5 r r6 r: ravelrs r r rH ranger7 transposer prodrC r r; r8 rd listrF ) r1 r@ r^ r_ _axisrestnewshaper rF r r r s @@r" vector_normr s w&&&HIII
A |
GGII D%
/tQV<<JJJJaf
JJJJJ$; LH % % - -
W////$///s
;
;
; Y>X>X>X>XSW>X>X>X Y Y
[
[
*RY^^AEs^;;
<
r s " " " " " " - , , , , , / / / / / / JIIIIIIIIIIIIIIII
z
J
,1
35 A A A A A A$
/
/
/
/ ,- P P P P P P
B
B
B
B 4 4 4 4 / / / / 7 7 7 7( 27rw [ [ [ [ [ [ ; ; ; ; GK : : : : : :05 5 5 5 6 6 6 6$ @D < < < < < <" ?H
I
I
I
I
I
I
H
H
H
H00 0 06 4 4 4 4 /3
]
]
]
]
]
]"A A A A cd E E E E E E )*D f f f f f f* 46 & & & & & &, OSej PQ + + + + + +Z F F Fr!