U ;v f`@sddlmZddlmZmZmZmZmZmZddl m Z ddl Z ddddd Z dddd d Zdddd d ZddddddZdddddZdddddZdddddZddddddZdddddZddddddZddddddZddddd Zddddd!d"Zddddd#d$Zddddd%d&Zdddd'd(Zdddd)d*Zdddd+d,Zddddd-d.Zddddd/d0Z dddd1d2Z!dddd3d4Z"dddd5d6Z#ddddd7d8Z$ddddd9d:Z%ddddd;d<Z&dddd=d>Z'dddd?d@Z(ddddAdBZ)dddddCdDZ*dddddEdFZ+ddddGdHZ,ddddIdJZ-ddddKdLZ.ddddMdNZ/dddddOdPZ0dddddQdRZ1ddddSdTZ2dddddUdVZ3dddddWdXZ4dddddYdZZ5dddd[d\Z6ddddd]d^Z7dddd_d`Z8dddddadbZ9dddddcddZ:ddddedfZ;ddddgdhZddddmdnZ?ddddodpZ@dddddqdrZAddddsdtZBddddudvZCddddwdxZDdS)y) annotations)_boolean_dtypes_floating_dtypes_integer_dtypes_integer_or_boolean_dtypes_numeric_dtypes _result_type)ArrayNr )xreturncCs$|jtkrtdtt|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_arrayr rJ/tmp/pip-unpacked-wheel-sdaoq5e2/numpy/array_api/_elementwise_functions.pyrs rcCs$|jtkrtdtt|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)r rrr rrZarccosrrrrracoss rcCs$|jtkrtdtt|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)r rrr rrZarccoshrrrrracosh(s r)x1x2r cCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r _normalize_two_argsrraddrrrrrrr3s rcCs$|jtkrtdtt|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)r rrr rrZarcsinrrrrrasinBs rcCs$|jtkrtdtt|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)r rrr rrZarcsinhrrrrrasinhNs rcCs$|jtkrtdtt|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)r rrr rrZarctanrrrrratanZs rcCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)z~ Array API compatible wrapper for :py:func:`np.arctan2 `. See its docstring for more information. z/Only floating-point dtypes are allowed in atan2) r rrr r rrrZarctan2rrrrratan2fs r cCs$|jtkrtdtt|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)r rrr rrZarctanhrrrrratanhus r!cCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrr bitwise_andrrrrrr"sr"cCsh|jtks|jtkrtdt|j|jt||\}}t|jdkrRt dt t |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) r rrr r rranyr ValueErrorrZ left_shiftrrrrbitwise_left_shiftsr%cCs$|jtkrtdtt|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) r rrr r rrr bitwise_orrrrrrr(sr(cCsh|jtks|jtkrtdt|j|jt||\}}t|jdkrRt dt t |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) r rrr r rrr#rr$rZ right_shiftrrrrbitwise_right_shiftsr)cCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrr bitwise_xorrrrrrr*sr*cCs2|jtkrtd|jtkr |Stt|jS)zx Array API compatible wrapper for :py:func:`np.ceil `. See its docstring for more information. z'Only numeric dtypes are allowed in ceil) r rrrr rrceilrrrrrr+s   r+cCs$|jtkrtdtt|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)r rrr rrcosrrrrrr,s r,cCs$|jtkrtdtt|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)r rrr rrcoshrrrrrr-s r-cCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrrdividerrrrrr.s r.cCs4t|j|jt||\}}tt|j|jS)zz Array API compatible wrapper for :py:func:`np.equal `. See its docstring for more information. )r r r rrrequalrrrrrr/sr/cCs$|jtkrtdtt|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)r rrr rrexprrrrrr0!s r0cCs$|jtkrtdtt|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)r rrr rrexpm1rrrrrr1,s r1cCs2|jtkrtd|jtkr |Stt|jS)zz Array API compatible wrapper for :py:func:`np.floor `. See its docstring for more information. z(Only numeric dtypes are allowed in floor) r rrrr rrfloorrrrrrr27s   r2cCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)z Array API compatible wrapper for :py:func:`np.floor_divide `. See its docstring for more information. z/Only numeric dtypes are allowed in floor_divide) r rrr r rrr floor_dividerrrrrr3Es r3cCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)z~ Array API compatible wrapper for :py:func:`np.greater `. See its docstring for more information. z*Only numeric dtypes are allowed in greater) r rrr r rrrgreaterrrrrrr4Ss r4cCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)z Array API compatible wrapper for :py:func:`np.greater_equal `. See its docstring for more information. z0Only numeric dtypes are allowed in greater_equal) r rrr r rrr greater_equalrrrrrr5as r5cCs$|jtkrtdtt|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)r rrr rrisfiniterrrrrr6os r6cCs$|jtkrtdtt|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)r rrr rrisinfrrrrrr7zs r7cCs$|jtkrtdtt|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)r rrr rrisnanrrrrrr8s r8cCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)zx Array API compatible wrapper for :py:func:`np.less `. See its docstring for more information. z'Only numeric dtypes are allowed in less) r rrr r rrrlessrrrrrr9s r9cCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)z Array API compatible wrapper for :py:func:`np.less_equal `. See its docstring for more information. z-Only numeric dtypes are allowed in less_equal) r rrr r rrr less_equalrrrrrr:s r:cCs$|jtkrtdtt|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)r rrr rrlogrrrrrr;s r;cCs$|jtkrtdtt|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)r rrr rrlog1prrrrrr<s r<cCs$|jtkrtdtt|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)r rrr rrlog2rrrrrr=s r=cCs$|jtkrtdtt|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)r rrr rrlog10rrrrrr>s r>cCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)z Array API compatible wrapper for :py:func:`np.logaddexp `. See its docstring for more information. z3Only floating-point dtypes are allowed in logaddexp) r rrr r rrr logaddexprrrrrr?s r?cCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrr logical_andrrrrrr@s r@cCs$|jtkrtdtt|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)r rrr rr logical_notrrrrrrAs rAcCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrr logical_orrrrrrrBs rBcCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrr logical_xorrrrrrrC s rCcCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrrmultiplyrrrrrrDs rDcCs$|jtkrtdtt|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)r rrr rrnegativerrrrrrE)s rEcCs4t|j|jt||\}}tt|j|jS)z Array API compatible wrapper for :py:func:`np.not_equal `. See its docstring for more information. )r r r rrr not_equalrrrrrrF4srFcCs$|jtkrtdtt|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)r rrr rrpositiverrrrrrG@s rGcCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrrpowerrrrrrpowLs rIcCsP|jtks|jtkrtdt|j|jt||\}}tt|j |j S)z Array API compatible wrapper for :py:func:`np.remainder `. See its docstring for more information. z,Only numeric dtypes are allowed in remainder) r rrr r rrr remainderrrrrrrJZs rJcCs$|jtkrtdtt|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)r rrr rrroundrrrrrrKhs rKcCs$|jtkrtdtt|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)r rrr rrsignrrrrrrLss rLcCs$|jtkrtdtt|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)r rrr rrsinrrrrrrM~s rMcCs$|jtkrtdtt|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)r rrr rrsinhrrrrrrNs rNcCs$|jtkrtdtt|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)r rrr rrsquarerrrrrrOs rOcCs$|jtkrtdtt|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)r rrr rrsqrtrrrrrrPs rPcCsP|jtks|jtkrtdt|j|jt||\}}tt|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 rrr r rrrsubtractrrrrrrQs rQcCs$|jtkrtdtt|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)r rrr rrtanrrrrrrRs rRcCs$|jtkrtdtt|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)r rrr rrtanhrrrrrrSs rScCs2|jtkrtd|jtkr |Stt|jS)zz Array API compatible wrapper for :py:func:`np.trunc `. See its docstring for more information. z(Only numeric dtypes are allowed in trunc) r rrrr rrtruncrrrrrrTs   rT)E __future__rZ_dtypesrrrrrr Z _array_objectr Znumpyrrrrrrrrr r!r"r%r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrIrJrKrLrMrNrOrPrQrRrSrTrrrrsv