0ԉ2 * d dl mZ ddlmZmZmZmZmZmZm Z m
Z
m Z ddl m
Z
d dlZdFd ZdFd
ZdFd ZdGdZdFdZdFdZdFdZdGdZdFdZdGdZdGdZdFdZdGdZ dGdZ dGdZ dFdZ dFdZ dFd Z!dFd Z"dGd Z#dGd Z$dFd Z%dFd!Z&dFd"Z'dGd#Z(dGd$Z)dGd%Z*dFd&Z+dFd'Z,dFd(Z-dFd)Z.dGd*Z/dGd+Z0dFd,Z1dFd-Z2dFd.Z3dFd/Z4dGd0Z5dGd1Z6dFd2Z7dGd3Z8dGd4Z9dGd5Z:dFd6Z;dGd7ZdFd:Z?dGd;Z@dFd<ZAdFd=ZBdFd>ZCdFd?ZDdFd@ZEdFdAZFdGdBZGdFdCZHdFdDZIdFdEZJdS )H ) annotations ) _boolean_dtypes_floating_dtypes_real_floating_dtypes_complex_floating_dtypes_integer_dtypes_integer_or_boolean_dtypes_real_numeric_dtypes_numeric_dtypes _result_type)ArrayNxr returnc | j t vrt d t j t j | j S )zv
Array API compatible wrapper for :py:func:`np.abs `.
See its docstring for more information.
z&Only numeric dtypes are allowed in abs)dtyper TypeErrorr _newnpabs_arrayr s z/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/array_api/_elementwise_functions.pyr r s< wo%%@AAA :bfQX&& ' '' c | j t vrt d t j t j | j S )z|
Array API compatible wrapper for :py:func:`np.arccos `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in acos)r r r r r r arccosr r s r acosr = w&&&HIII :bi)) * **r c | j t vrt d t j t j | j S )z~
Array API compatible wrapper for :py:func:`np.arccosh `.
See its docstring for more information.
z/Only floating-point dtypes are allowed in acosh)r r r r r r arccoshr r s r acoshr! + = w&&&IJJJ :bj** + ++r x1x2c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )zv
Array API compatible wrapper for :py:func:`np.add `.
See its docstring for more information.
z&Only numeric dtypes are allowed in add)
r r r r
r _normalize_two_argsr r addr r# r$ s r r' r' 6 sv
x&&"(/*I*I@AAA28$$$
&r2
.
.FB :bfRY 22 3 33r c | j t vrt d t j t j | j S )z|
Array API compatible wrapper for :py:func:`np.arcsin `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in asin)r r r r r r arcsinr r s r asinr+ E r r c | j t vrt d t j t j | j S )z~
Array API compatible wrapper for :py:func:`np.arcsinh `.
See its docstring for more information.
z/Only floating-point dtypes are allowed in asinh)r r r r r r arcsinhr r s r asinhr. Q r" r c | j t vrt d t j t j | j S )z|
Array API compatible wrapper for :py:func:`np.arctan `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in atan)r r r r r r arctanr r s r atanr1 ] r r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z~
Array API compatible wrapper for :py:func:`np.arctan2 `.
See its docstring for more information.
z4Only real floating-point dtypes are allowed in atan2)
r r r r
r r& r r arctan2r r( s r atan2r4 i sx
x,,,@U0U0UNOOO28$$$
&r2
.
.FB :bjBI66 7 77r c | j t vrt d t j t j | j S )z~
Array API compatible wrapper for :py:func:`np.arctanh `.
See its docstring for more information.
z/Only floating-point dtypes are allowed in atanh)r r r r r r arctanhr r s r atanhr7 x r" r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.bitwise_and `.
See its docstring for more information.
z9Only integer or boolean dtypes are allowed in bitwise_and)
r r
r r
r r& r r bitwise_andr r( s r r9 r9 y 222
85 5 5STTT28$$$
&r2
.
.FB :bnRY :: ; ;;r c v | j t vs|j t vrt d t | j |j t j | | \ } }t
j |j dk rt d t j
t
j | j |j S )z
Array API compatible wrapper for :py:func:`np.left_shift `.
See its docstring for more information.
z5Only integer dtypes are allowed in bitwise_left_shiftr z6bitwise_left_shift(x1, x2) is only defined for x2 >= 0) r r r r
r r& r anyr
ValueErrorr
left_shiftr( s r bitwise_left_shiftr? s
x&&"(/*I*IOPPP28$$$
&r2
.
.FB vbi!m SQRRR :bmBIry99 : ::r c | j t vrt d t j t j | j S )z|
Array API compatible wrapper for :py:func:`np.invert `.
See its docstring for more information.
z`.
See its docstring for more information.
z8Only integer or boolean dtypes are allowed in bitwise_or)
r r
r r
r r& r r
bitwise_orr r( s r rD rD sy 222
85 5 5RSSS28$$$
&r2
.
.FB :bmBIry99 : ::r c v | j t vs|j t vrt d t | j |j t j | | \ } }t
j |j dk rt d t j
t
j | j |j S )z
Array API compatible wrapper for :py:func:`np.right_shift `.
See its docstring for more information.
z6Only integer dtypes are allowed in bitwise_right_shiftr z7bitwise_right_shift(x1, x2) is only defined for x2 >= 0) r r r r
r r& r r< r r= r right_shiftr( s r bitwise_right_shiftrG s
x&&"(/*I*IPQQQ28$$$
&r2
.
.FB vbi!m TRSSS :bnRY :: ; ;;r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.bitwise_xor `.
See its docstring for more information.
z9Only integer or boolean dtypes are allowed in bitwise_xor)
r r
r r
r r& r r bitwise_xorr r( s r rI rI r: r c | j t vrt d | j t v r| S t j t
j | j S )zx
Array API compatible wrapper for :py:func:`np.ceil `.
See its docstring for more information.
z,Only real numeric dtypes are allowed in ceil) r r r r r r r ceilr r s r rK rK sO w***FGGGw/!! :bgah'' ( ((r c | j t vrt d t j t j | S )zx
Array API compatible wrapper for :py:func:`np.conj `.
See its docstring for more information.
z6Only complex floating-point dtypes are allowed in conj)r r r r r r conjr s r rM rM 9 w...PQQQ :bgajj ! !!r c | j t vrt d t j t j | j S )zv
Array API compatible wrapper for :py:func:`np.cos `.
See its docstring for more information.
z-Only floating-point dtypes are allowed in cos)r r r r r r cosr r s r rP rP = w&&&GHHH :bfQX&& ' ''r c | j t vrt d t j t j | j S )zx
Array API compatible wrapper for :py:func:`np.cosh `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in cosh)r r r r r r coshr r s r rS rS
= w&&&HIII :bgah'' ( ((r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z|
Array API compatible wrapper for :py:func:`np.divide `.
See its docstring for more information.
z0Only floating-point dtypes are allowed in divide)
r r r r
r r& r r divider r( s r rV rV sx
x'''28;K+K+KJKKK28$$$
&r2
.
.FB :bi 2955 6 66r c t | j |j t j | | \ } }t j t j | j |j S )zz
Array API compatible wrapper for :py:func:`np.equal `.
See its docstring for more information.
)r
r r r& r r equalr r( s r rX rX # sN 28$$$
&r2
.
.FB :bhry")44 5 55r c | j t vrt d t j t j | j S )zv
Array API compatible wrapper for :py:func:`np.exp `.
See its docstring for more information.
z-Only floating-point dtypes are allowed in exp)r r r r r r expr r s r rZ rZ / rQ r c | j t vrt d t j t j | j S )zz
Array API compatible wrapper for :py:func:`np.expm1 `.
See its docstring for more information.
z/Only floating-point dtypes are allowed in expm1)r r r r r r expm1r r s r r\ r\ : = w&&&IJJJ :bhqx(( ) ))r c | j t vrt d | j t v r| S t j t
j | j S )zz
Array API compatible wrapper for :py:func:`np.floor `.
See its docstring for more information.
z-Only real numeric dtypes are allowed in floor) r r r r r r r floorr r s r r_ r_ E O w***GHHHw/!! :bhqx(( ) ))r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.floor_divide `.
See its docstring for more information.
z4Only real numeric dtypes are allowed in floor_divide)
r r r r
r r& r r floor_divider r( s r rb rb S sx
x+++rx?S/S/SNOOO28$$$
&r2
.
.FB :bobi;; < <`.
See its docstring for more information.
z/Only real numeric dtypes are allowed in greater)
r r r r
r r& r r greaterr r( s r rd rd a sx
x+++rx?S/S/SIJJJ28$$$
&r2
.
.FB :bjBI66 7 77r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.greater_equal `.
See its docstring for more information.
z5Only real numeric dtypes are allowed in greater_equal)
r r r r
r r& r r
greater_equalr r( s r rf rf o sy
x+++rx?S/S/SOPPP28$$$
&r2
.
.FB :b&ry")<< = ==r c | j t vrt d t j t j | S )zx
Array API compatible wrapper for :py:func:`np.imag `.
See its docstring for more information.
z6Only complex floating-point dtypes are allowed in imag)r r r r r r imagr s r rh rh } rN r c | j t vrt d t j t j | j S )z
Array API compatible wrapper for :py:func:`np.isfinite `.
See its docstring for more information.
z+Only numeric dtypes are allowed in isfinite)r r r r r r isfiniter r s r rj rj < wo%%EFFF :bk!(++ , ,,r c | j t vrt d t j t j | j S )zz
Array API compatible wrapper for :py:func:`np.isinf `.
See its docstring for more information.
z(Only numeric dtypes are allowed in isinf)r r r r r r isinfr r s r rm rm < wo%%BCCC :bhqx(( ) ))r c | j t vrt d t j t j | j S )zz
Array API compatible wrapper for :py:func:`np.isnan `.
See its docstring for more information.
z(Only numeric dtypes are allowed in isnan)r r r r r r isnanr r s r rp rp rn r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )zx
Array API compatible wrapper for :py:func:`np.less `.
See its docstring for more information.
z,Only real numeric dtypes are allowed in less)
r r r r
r r& r r lessr r( s r rr rr sx
x+++rx?S/S/SFGGG28$$$
&r2
.
.FB :bgbi33 4 44r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.less_equal `.
See its docstring for more information.
z2Only real numeric dtypes are allowed in less_equal)
r r r r
r r& r r
less_equalr r( s r rt rt sx
x+++rx?S/S/SLMMM28$$$
&r2
.
.FB :bmBIry99 : ::r c | j t vrt d t j t j | j S )zv
Array API compatible wrapper for :py:func:`np.log `.
See its docstring for more information.
z-Only floating-point dtypes are allowed in log)r r r r r r logr r s r rv rv rQ r c | j t vrt d t j t j | j S )zz
Array API compatible wrapper for :py:func:`np.log1p `.
See its docstring for more information.
z/Only floating-point dtypes are allowed in log1p)r r r r r r log1pr r s r rx rx r] r c | j t vrt d t j t j | j S )zx
Array API compatible wrapper for :py:func:`np.log2 `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in log2)r r r r r r log2r r s r rz rz rT r c | j t vrt d t j t j | j S )zz
Array API compatible wrapper for :py:func:`np.log10 `.
See its docstring for more information.
z/Only floating-point dtypes are allowed in log10)r r r r r r log10r r s r r| r| r] r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.logaddexp `.
See its docstring for more information.
z8Only real floating-point dtypes are allowed in logaddexp)
r r r r
r r& r r logaddexpr r( s r r~ r~ sx
x,,,@U0U0URSSS28$$$
&r2
.
.FB :bl29bi88 9 99r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.logical_and `.
See its docstring for more information.
z.Only boolean dtypes are allowed in logical_and)
r r r r
r r& r r logical_andr r( s r r r v
x&&"(/*I*IHIII28$$$
&r2
.
.FB :bnRY :: ; ;;r c | j t vrt d t j t j | j S )z
Array API compatible wrapper for :py:func:`np.logical_not `.
See its docstring for more information.
z.Only boolean dtypes are allowed in logical_not)r r r r r r logical_notr r s r r r
s< wo%%HIII :bnQX.. / //r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.logical_or `.
See its docstring for more information.
z-Only boolean dtypes are allowed in logical_or)
r r r r
r r& r r
logical_orr r( s r r r sv
x&&"(/*I*IGHHH28$$$
&r2
.
.FB :bmBIry99 : ::r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.logical_xor `.
See its docstring for more information.
z.Only boolean dtypes are allowed in logical_xor)
r r r r
r r& r r logical_xorr r( s r r r & r r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.multiply `.
See its docstring for more information.
z+Only numeric dtypes are allowed in multiply)
r r r r
r r& r r multiplyr r( s r r r 4 v
x&&"(/*I*IEFFF28$$$
&r2
.
.FB :bk")RY77 8 88r c | j t vrt d t j t j | j S )z
Array API compatible wrapper for :py:func:`np.negative `.
See its docstring for more information.
z+Only numeric dtypes are allowed in negative)r r r r r r negativer r s r r r B rk r c t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.not_equal `.
See its docstring for more information.
)r
r r r& r r not_equalr r( s r r r M sN 28$$$
&r2
.
.FB :bl29bi88 9 99r c | j t vrt d t j t j | j S )z
Array API compatible wrapper for :py:func:`np.positive `.
See its docstring for more information.
z+Only numeric dtypes are allowed in positive)r r r r r r positiver r s r r r Y rk r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )zz
Array API compatible wrapper for :py:func:`np.power `.
See its docstring for more information.
z&Only numeric dtypes are allowed in pow)
r r r r
r r& r r powerr r( s r powr e sv
x&&"(/*I*I@AAA28$$$
&r2
.
.FB :bhry")44 5 55r c | j t vrt d t j t j | S )zx
Array API compatible wrapper for :py:func:`np.real `.
See its docstring for more information.
z6Only complex floating-point dtypes are allowed in real)r r r r r r realr s r r r s rN r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.remainder `.
See its docstring for more information.
z1Only real numeric dtypes are allowed in remainder)
r r r r
r r& r r remainderr r( s r r r ~ sx
x+++rx?S/S/SKLLL28$$$
&r2
.
.FB :bl29bi88 9 99r c | j t vrt d t j t j | j S )zz
Array API compatible wrapper for :py:func:`np.round `.
See its docstring for more information.
z(Only numeric dtypes are allowed in round)r r r r r r roundr r s r r r rn r c | j t vrt d t j t j | j S )zx
Array API compatible wrapper for :py:func:`np.sign `.
See its docstring for more information.
z'Only numeric dtypes are allowed in sign)r r r r r r signr r s r r r s< wo%%ABBB :bgah'' ( ((r c | j t vrt d t j t j | j S )zv
Array API compatible wrapper for :py:func:`np.sin `.
See its docstring for more information.
z-Only floating-point dtypes are allowed in sin)r r r r r r sinr r s r r r rQ r c | j t vrt d t j t j | j S )zx
Array API compatible wrapper for :py:func:`np.sinh `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in sinh)r r r r r r sinhr r s r r r rT r c | j t vrt d t j t j | j S )z|
Array API compatible wrapper for :py:func:`np.square `.
See its docstring for more information.
z)Only numeric dtypes are allowed in square)r r r r r r squarer r s r r r s< wo%%CDDD :bi)) * **r c | j t vrt d t j t j | j S )zx
Array API compatible wrapper for :py:func:`np.sqrt `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in sqrt)r r r r r r sqrtr r s r r r rT r c | j t vs|j t vrt d t | j |j t j | | \ } }t j t j | j |j S )z
Array API compatible wrapper for :py:func:`np.subtract `.
See its docstring for more information.
z+Only numeric dtypes are allowed in subtract)
r r r r
r r& r r subtractr r( s r r r r r c | j t vrt d t j t j | j S )zv
Array API compatible wrapper for :py:func:`np.tan `.
See its docstring for more information.
z-Only floating-point dtypes are allowed in tan)r r r r r r tanr r s r r r rQ r c | j t vrt d t j t j | j S )zx
Array API compatible wrapper for :py:func:`np.tanh `.
See its docstring for more information.
z.Only floating-point dtypes are allowed in tanh)r r r r r r tanhr r s r r r rT r c | j t vrt d | j t v r| S t j t
j | j S )zz
Array API compatible wrapper for :py:func:`np.trunc `.
See its docstring for more information.
z-Only real numeric dtypes are allowed in trunc) r r r r r r r truncr r s r r r r` r )r r r r )r# r r$ r r r )K
__future__r _dtypesr r r r r r
r r r
_array_objectr numpyr r r r! r' r+ r. r1 r4 r7 r9 r? rB rD rG rI rK rM rP rS rV rX rZ r\ r_ rb rd rf rh rj rm rp rr rt rv rx rz r| r~ r r r r r r r r r r r r r r r r r r r r r r r r sK " " " " " "
! ( ( ( (+ + + +, , , , 4 4 4 4 + + + +, , , ,+ + + + 8 8 8 8 , , , ,< < < <$; ; ; ;$+ + + +; ; ; ;$< < < <"< < < <" ) ) ) ) " " " "( ( ( () ) ) ) 7 7 7 7 6 6 6 6( ( ( (* * * * * * * * = = = = 8 8 8 8 > > > > " " " "- - - -* * * ** * * * 5 5 5 5 ; ; ; ; ( ( ( (* * * *) ) ) )* * * * : : : : < < < < 0 0 0 0 ; ; ; ; < < < < 9 9 9 9 - - - - : : : :- - - - 6 6 6 6 " " " " : : : : * * * *) ) ) )( ( ( () ) ) )+ + + +) ) ) ) 9 9 9 9 ( ( ( () ) ) ) * * * * * *r