a ;¹§eˆeã@s.ddlmZddlmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlZdddœdd „Zdddœd d „Zdddœd d „Zddddœdd„Zdddœdd„Zdddœdd„Zdddœdd„Zddddœdd„Zdddœdd„Zddddœdd„Zddddœdd„Zdddœdd „Zddddœd!d"„Zddddœd#d$„Zddddœd%d&„Zdddœd'd(„Zdddœd)d*„Z dddœd+d,„Z!dddœd-d.„Z"ddddœd/d0„Z#ddddœd1d2„Z$dddœd3d4„Z%dddœd5d6„Z&dddœd7d8„Z'ddddœd9d:„Z(ddddœd;d<„Z)ddddœd=d>„Z*dddœd?d@„Z+dddœdAdB„Z,dddœdCdD„Z-dddœdEdF„Z.ddddœdGdH„Z/ddddœdIdJ„Z0dddœdKdL„Z1dddœdMdN„Z2dddœdOdP„Z3dddœdQdR„Z4ddddœdSdT„Z5ddddœdUdV„Z6dddœdWdX„Z7ddddœdYdZ„Z8ddddœd[d\„Z9ddddœd]d^„Z:dddœd_d`„Z;ddddœdadb„Zdddœdgdh„Z?ddddœdidj„Z@dddœdkdl„ZAdddœdmdn„ZBdddœdodp„ZCdddœdqdr„ZDdddœdsdt„ZEdddœdudv„ZFddddœdwdx„ZGdddœdydz„ZHdddœd{d|„ZIdddœd}d~„ZJdS)é)Ú annotationsé) Ú_boolean_dtypesÚ_floating_dtypesÚ_real_floating_dtypesÚ_complex_floating_dtypesÚ_integer_dtypesÚ_integer_or_boolean_dtypesÚ_real_numeric_dtypesÚ_numeric_dtypesÚ _result_type)ÚArrayNr )ÚxÚreturncCs$|jtvrtdƒ‚t t |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 Ú_newÚnpÚabsÚ_array©r©rúZ/var/www/sunrise/env/lib/python3.9/site-packages/numpy/array_api/_elementwise_functions.pyrs rcCs$|jtvrtdƒ‚t t |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)rrrr rrZarccosrrrrrÚacoss rcCs$|jtvrtdƒ‚t t |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)rrrr rrZarccoshrrrrrÚacosh+s r)Úx1Úx2rcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r Ú_normalize_two_argsrrÚaddr©rrrrrr6s rcCs$|jtvrtdƒ‚t t |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)rrrr rrZarcsinrrrrrÚasinEs r cCs$|jtvrtdƒ‚t t |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)rrrr rrZarcsinhrrrrrÚasinhQs r!cCs$|jtvrtdƒ‚t t |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)rrrr rrZarctanrrrrrÚatan]s r"cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rrrr r rrrZarctan2rrrrrÚatan2is r#cCs$|jtvrtdƒ‚t t |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)rrrr rrZarctanhrrrrrÚatanhxs r$cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚ bitwise_andrrrrrr%ƒsÿþr%cCsh|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t |jdk¡rRt dƒ‚t  t  |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_shiftrz6bitwise_left_shift(x1, x2) is only defined for x2 >= 0) rrrr r rrÚanyrÚ ValueErrorrZ left_shiftrrrrÚbitwise_left_shift•sr(cCs$|jtvrtdƒ‚t t |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) rr rr r rrrÚ bitwise_orrrrrrr+²sÿþr+cCsh|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t |jdk¡rRt dƒ‚t  t  |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_shiftrz7bitwise_right_shift(x1, x2) is only defined for x2 >= 0) rrrr r rrr&rr'rZ right_shiftrrrrÚbitwise_right_shiftÄsr,cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚ bitwise_xorrrrrrr-Õsÿþr-cCs2|jtvrtdƒ‚|jtvr |St t |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) rr rrr rrÚceilrrrrrr.æs   r.cCs"|jtvrtdƒ‚t t |¡¡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)rrrr rrÚconjrrrrr/ôs r/cCs$|jtvrtdƒ‚t t |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)rrrr rrÚcosrrrrrr0ÿs r0cCs$|jtvrtdƒ‚t t |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)rrrr rrÚcoshrrrrrr1 s r1cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rrrr r rrrÚdividerrrrrr2s r2cCs4t|j|jƒt ||¡\}}t t |j|j¡¡S)zz Array API compatible wrapper for :py:func:`np.equal `. See its docstring for more information. )r rr rrrÚequalrrrrrr3#sr3cCs$|jtvrtdƒ‚t t |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)rrrr rrÚexprrrrrr4/s r4cCs$|jtvrtdƒ‚t t |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)rrrr rrÚexpm1rrrrrr5:s r5cCs2|jtvrtdƒ‚|jtvr |St t |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) rr rrr rrÚfloorrrrrrr6Es   r6cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚ floor_dividerrrrrr7Ss r7cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |j |j ¡¡S)z~ Array API compatible wrapper for :py:func:`np.greater `. See its docstring for more information. z/Only real numeric dtypes are allowed in greater) rr rr r rrrÚgreaterrrrrrr8as r8cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚ greater_equalrrrrrr9os r9cCs"|jtvrtdƒ‚t t |¡¡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)rrrr rrÚimagrrrrr:}s r:cCs$|jtvrtdƒ‚t t |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)rr rr rrÚisfiniterrrrrr;ˆs r;cCs$|jtvrtdƒ‚t t |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)rr rr rrÚisinfrrrrrr<“s r<cCs$|jtvrtdƒ‚t t |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)rr rr rrÚisnanrrrrrr=žs r=cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚlessrrrrrr>©s r>cCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚ less_equalrrrrrr?·s r?cCs$|jtvrtdƒ‚t t |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)rrrr rrÚlogrrrrrr@Ås r@cCs$|jtvrtdƒ‚t t |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)rrrr rrÚlog1prrrrrrAÐs rAcCs$|jtvrtdƒ‚t t |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)rrrr rrÚlog2rrrrrrBÛs rBcCs$|jtvrtdƒ‚t t |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)rrrr rrÚlog10rrrrrrCæs rCcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rrrr r rrrÚ logaddexprrrrrrDñs rDcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rrrr r rrrÚ logical_andrrrrrrEÿs rEcCs$|jtvrtdƒ‚t t |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)rrrr rrÚ logical_notrrrrrrF s rFcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rrrr r rrrÚ logical_orrrrrrrGs rGcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rrrr r rrrÚ logical_xorrrrrrrH&s rHcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚmultiplyrrrrrrI4s rIcCs$|jtvrtdƒ‚t t |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)rr rr rrÚnegativerrrrrrJBs rJcCs4t|j|jƒt ||¡\}}t t |j|j¡¡S)z‚ Array API compatible wrapper for :py:func:`np.not_equal `. See its docstring for more information. )r rr rrrÚ not_equalrrrrrrKMsrKcCs$|jtvrtdƒ‚t t |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)rr rr rrÚpositiverrrrrrLYs rLcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚpowerrrrrrÚpowes rNcCs"|jtvrtdƒ‚t t |¡¡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)rrrr rrÚrealrrrrrOss rOcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚ remainderrrrrrrP~s rPcCs$|jtvrtdƒ‚t t |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)rr rr rrÚroundrrrrrrQŒs rQcCs$|jtvrtdƒ‚t t |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)rr rr rrÚsignrrrrrrR—s rRcCs$|jtvrtdƒ‚t t |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)rrrr rrÚsinrrrrrrS¢s rScCs$|jtvrtdƒ‚t t |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)rrrr rrÚsinhrrrrrrT­s rTcCs$|jtvrtdƒ‚t t |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)rr rr rrÚsquarerrrrrrU¸s rUcCs$|jtvrtdƒ‚t t |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)rrrr rrÚsqrtrrrrrrVÃs rVcCsP|jtvs|jtvrtdƒ‚t|j|jƒt ||¡\}}t t |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) rr rr r rrrÚsubtractrrrrrrWÎs rWcCs$|jtvrtdƒ‚t t |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)rrrr rrÚtanrrrrrrXÜs rXcCs$|jtvrtdƒ‚t t |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)rrrr rrÚtanhrrrrrrYçs rYcCs2|jtvrtdƒ‚|jtvr |St t |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) rr rrr rrÚtruncrrrrrrZòs   rZ)KÚ __future__rZ_dtypesrrrrrr r r r Z _array_objectr Únumpyrrrrrr r!r"r#r$r%r(r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrNrOrPrQrRrSrTrUrVrWrXrYrZrrrrÚs| ,