{h-ih(ddlmZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlZdAdZdAdZdAdZdBd ZdAd ZdAd ZdAd ZdBd ZdAdZdBdZdBdZdAdZdBdZdBdZdBdZdAdZdAdZ dAdZ!dAdZ"dBdZ#dBdZ$dAdZ%dAdZ&dAdZ'dBdZ(dBdZ)dBd Z*dAd!Z+dAd"Z,dAd#Z-dAd$Z.dBd%Z/dBd&Z0dAd'Z1dAd(Z2dAd)Z3dAd*Z4dBd+Z5dBd,Z6dAd-Z7dBd.Z8dBd/Z9dBd0Z:dAd1Z;dBd2ZdAd5Z?dBd6Z@dAd7ZAdAd8ZBdAd9ZCdAd:ZDdAd;ZEdAd<ZFdBd=ZGdAd>ZHdAd?ZIdAd@ZJy)C) annotations) _boolean_dtypes_floating_dtypes_real_floating_dtypes_complex_floating_dtypes_integer_dtypes_integer_or_boolean_dtypes_real_numeric_dtypes_numeric_dtypes _result_type)ArrayNc|jtvr tdtjt j |jS)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_arrayxs eD:\DhanshreeandTeamAI\Github\ai_code\venv\Lib\site-packages\numpy/array_api/_elementwise_functions.pyrrs8  wwo%@AA ::bffQXX& ''c|jtvr tdtjt j |jS)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)rrrrrrarccosrrs racosr9  ww&&HII ::bii) **rc|jtvr tdtjt j |jS)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)rrrrrrarccoshrrs racoshr +9  ww&&IJJ ::bjj* ++rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)zv Array API compatible wrapper for :py:func:`np.add `. See its docstring for more information. z&Only numeric dtypes are allowed in add) rr rr r_normalize_two_argsrraddrx1x2s rr$r$6sq  xx&"((/*I@AA288$  & &r2 .FB ::bffRYY 2 33rc|jtvr tdtjt j |jS)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)rrrrrrarcsinrrs rasinr*Errc|jtvr tdtjt j |jS)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)rrrrrrarcsinhrrs rasinhr-Qr!rc|jtvr tdtjt j |jS)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)rrrrrrarctanrrs ratanr0]rrcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rrrr rr#rrarctan2rr%s ratan2r3iss  xx,,@U0UNOO288$  & &r2 .FB ::bjjBII6 77rc|jtvr tdtjt j |jS)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)rrrrrrarctanhrrs ratanhr6xr!rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rr rr rr#rr bitwise_andrr%s rr8r8t 22 885 5STT288$  & &r2 .FB ::bnnRYY : ;;rc|jtvs|jtvr tdt|j|jt j ||\}}t j|jdkr tdt jt j|j|jS)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_shiftrz6bitwise_left_shift(x1, x2) is only defined for x2 >= 0) rr rr rr#ranyr ValueErrorr left_shiftr%s rbitwise_left_shiftr>s  xx&"((/*IOPP288$  & &r2 .FB vvbii!mQRR ::bmmBIIryy9 ::rc|jtvr tdtjt j |jS)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) rr rr rr#rr bitwise_orrr%s rrCrCst 22 885 5RSS288$  & &r2 .FB ::bmmBIIryy9 ::rc|jtvs|jtvr tdt|j|jt j ||\}}t j|jdkr tdt jt j|j|jS)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_shiftrz7bitwise_right_shift(x1, x2) is only defined for x2 >= 0) rr rr rr#rr;rr<r right_shiftr%s rbitwise_right_shiftrFs  xx&"((/*IPQQ288$  & &r2 .FB vvbii!mRSS ::bnnRYY : ;;rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rr rr rr#rr bitwise_xorrr%s rrHrHr9rc|jtvr td|jtvr|St j t j|jS)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) rr rr rrrceilrrs rrJrJsJ  ww**FGGww/! ::bggahh' ((rc|jtvr tdtjt 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)rrrrrrconjrs rrLrL4  ww..PQQ ::bggaj !!rc|jtvr tdtjt j |jS)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)rrrrrrcosrrs rrOrO9  ww&&GHH ::bffQXX& ''rc|jtvr tdtjt j |jS)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)rrrrrrcoshrrs rrRrR 9  ww&&HII ::bggahh' ((rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)z| Array API compatible wrapper for :py:func:`np.divide `. See its docstring for more information. z0Only floating-point dtypes are allowed in divide) rrrr rr#rrdividerr%s rrUrUss  xx''288;K+KJKK288$  & &r2 .FB ::bii 2995 66rct|j|jtj||\}}tjt j |j|jS)zz Array API compatible wrapper for :py:func:`np.equal `. See its docstring for more information. )r rrr#rrequalrr%s rrWrW#sN288$  & &r2 .FB ::bhhryy"))4 55rc|jtvr tdtjt j |jS)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)rrrrrrexprrs rrYrY/rPrc|jtvr tdtjt j |jS)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)rrrrrrexpm1rrs rr[r[:9  ww&&IJJ ::bhhqxx( ))rc|jtvr td|jtvr|St j t j|jS)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) rr rr rrrfloorrrs rr^r^EJ  ww**GHHww/! ::bhhqxx( ))rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rr rr rr#rr floor_dividerr%s rraraSss  xx++rxx?S/SNOO288$  & &r2 .FB ::boobii; <`. See its docstring for more information. z/Only real numeric dtypes are allowed in greater) rr rr rr#rrgreaterrr%s rrcrcass  xx++rxx?S/SIJJ288$  & &r2 .FB ::bjjBII6 77rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rr rr rr#rr greater_equalrr%s rrereosu  xx++rxx?S/SOPP288$  & &r2 .FB ::b&&ryy"))< ==rc|jtvr tdtjt 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)rrrrrrimagrs rrgrg}rMrc|jtvr tdtjt j |jS)z Array API compatible wrapper for :py:func:`np.isfinite `. See its docstring for more information. z+Only numeric dtypes are allowed in isfinite)rr rrrrisfiniterrs rriri8  wwo%EFF ::bkk!((+ ,,rc|jtvr tdtjt j |jS)zz Array API compatible wrapper for :py:func:`np.isinf `. See its docstring for more information. z(Only numeric dtypes are allowed in isinf)rr rrrrisinfrrs rrlrl8  wwo%BCC ::bhhqxx( ))rc|jtvr tdtjt j |jS)zz Array API compatible wrapper for :py:func:`np.isnan `. See its docstring for more information. z(Only numeric dtypes are allowed in isnan)rr rrrrisnanrrs rrorormrcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rr rr rr#rrlessrr%s rrqrqss  xx++rxx?S/SFGG288$  & &r2 .FB ::bggbii3 44rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rr rr rr#rr less_equalrr%s rrsrsss  xx++rxx?S/SLMM288$  & &r2 .FB ::bmmBIIryy9 ::rc|jtvr tdtjt j |jS)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)rrrrrrlogrrs rrururPrc|jtvr tdtjt j |jS)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)rrrrrrlog1prrs rrwrwr\rc|jtvr tdtjt j |jS)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)rrrrrrlog2rrs rryryrSrc|jtvr tdtjt j |jS)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)rrrrrrlog10rrs rr{r{r\rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rrrr rr#rr logaddexprr%s rr}r}ss  xx,,@U0URSS288$  & &r2 .FB ::bll299bii8 99rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rrrr rr#rr logical_andrr%s rrrq  xx&"((/*IHII288$  & &r2 .FB ::bnnRYY : ;;rc|jtvr tdtjt j |jS)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)rrrrrr logical_notrrs rrr s8  wwo%HII ::bnnQXX. //rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rrrr rr#rr logical_orrr%s rrrsq  xx&"((/*IGHH288$  & &r2 .FB ::bmmBIIryy9 ::rcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)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) rrrr rr#rr logical_xorrr%s rrr&rrcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)z Array API compatible wrapper for :py:func:`np.multiply `. See its docstring for more information. z+Only numeric dtypes are allowed in multiply) rr rr rr#rrmultiplyrr%s rrr4q  xx&"((/*IEFF288$  & &r2 .FB ::bkk"))RYY7 88rc|jtvr tdtjt j |jS)z Array API compatible wrapper for :py:func:`np.negative `. See its docstring for more information. z+Only numeric dtypes are allowed in negative)rr rrrrnegativerrs rrrBrjrct|j|jtj||\}}tjt j |j|jS)z Array API compatible wrapper for :py:func:`np.not_equal `. See its docstring for more information. )r rrr#rr not_equalrr%s rrrMsN288$  & &r2 .FB ::bll299bii8 99rc|jtvr tdtjt j |jS)z Array API compatible wrapper for :py:func:`np.positive `. See its docstring for more information. z+Only numeric dtypes are allowed in positive)rr rrrrpositiverrs rrrYrjrcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)zz Array API compatible wrapper for :py:func:`np.power `. See its docstring for more information. z&Only numeric dtypes are allowed in pow) rr rr rr#rrpowerrr%s rpowresq  xx&"((/*I@AA288$  & &r2 .FB ::bhhryy"))4 55rc|jtvr tdtjt 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)rrrrrrrealrs rrrsrMrcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)z Array API compatible wrapper for :py:func:`np.remainder `. See its docstring for more information. z1Only real numeric dtypes are allowed in remainder) rr rr rr#rr remainderrr%s rrr~ss  xx++rxx?S/SKLL288$  & &r2 .FB ::bll299bii8 99rc|jtvr tdtjt j |jS)zz Array API compatible wrapper for :py:func:`np.round `. See its docstring for more information. z(Only numeric dtypes are allowed in round)rr rrrrroundrrs rrrrmrc|jtvr tdtjt j |jS)zx Array API compatible wrapper for :py:func:`np.sign `. See its docstring for more information. z'Only numeric dtypes are allowed in sign)rr rrrrsignrrs rrrs8  wwo%ABB ::bggahh' ((rc|jtvr tdtjt j |jS)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)rrrrrrsinrrs rrrrPrc|jtvr tdtjt j |jS)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)rrrrrrsinhrrs rrrrSrc|jtvr tdtjt j |jS)z| Array API compatible wrapper for :py:func:`np.square `. See its docstring for more information. z)Only numeric dtypes are allowed in square)rr rrrrsquarerrs rrrs8  wwo%CDD ::bii) **rc|jtvr tdtjt j |jS)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)rrrrrrsqrtrrs rrrrSrcL|jtvs|jtvr tdt|j|jt j ||\}}t j tj|j|jS)z Array API compatible wrapper for :py:func:`np.subtract `. See its docstring for more information. z+Only numeric dtypes are allowed in subtract) rr rr rr#rrsubtractrr%s rrrrrc|jtvr tdtjt j |jS)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)rrrrrrtanrrs rrrrPrc|jtvr tdtjt j |jS)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)rrrrrrtanhrrs rrrrSrc|jtvr td|jtvr|St j t j|jS)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) rr rr rrrtruncrrs rrrr_r)rrreturnr)r&rr'rrr)K __future__r_dtypesrrrrr r r r r _array_objectrnumpyrrrr r$r*r-r0r3r6r8r>rArCrFrHrJrLrOrRrUrWrYr[r^rarcrergrirlrorqrsrurwryr{r}rrrrrrrrrrrrrrrrrrrrrrrrsF"   !(+, 4+,+ 8,<$;$+;$<"<" )"() 7 6(* * = 8 >"-** 5 ;(*)* : <0 ; < 9- :- 6" :*)()+) 9() *r