g`ddlmZddlmZmZmZmZmZmZddl m Z ddl Z dCd Z dCd ZdCd ZdDdZdCdZdCdZdCdZdDdZdCdZdDdZdDdZdCdZdDdZdDdZdDdZdCdZdCdZdCdZdDdZdDdZ dCdZ!dCd Z"dCd!Z#dDd"Z$dDd#Z%dDd$Z&dCd%Z'dCd&Z(dCd'Z)dDd(Z*dDd)Z+dCd*Z,dCd+Z-dCd,Z.dCd-Z/dDd.Z0dDd/Z1dCd0Z2dDd1Z3dDd2Z4dDd3Z5dCd4Z6dDd5Z7dCd6Z8dDd7Z9dDd8Z:dCd9Z;dCd:ZdCd=Z?dCd>Z@dDd?ZAdCd@ZBdCdAZCdCdBZDdS)E) annotations)_boolean_dtypes_floating_dtypes_integer_dtypes_integer_or_boolean_dtypes_numeric_dtypes _result_type)ArrayNxr returnc|jtvrtdtjt 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_arrayr s \/granary/edev/quantum/lib/python3.11/site-packages/numpy/array_api/_elementwise_functions.pyrrs<  wo%%@AAA :bfQX&& ' ''c|jtvrtdtjt 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)rrrr rrarccosrrs racosr=  w&&&HIII :bi)) * **rc|jtvrtdtjt 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)rrrr rrarccoshrrs racoshr(=  w&&&IJJJ :bj** + ++rx1x2c|jtvs|jtvrtdt|j|jt j||\}}t jtj|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_argsrraddrr r!s rr$r$3sv  x&&"(/*I*I@AAA28$$$  &r2 . .FB :bfRY 22 3 33rc|jtvrtdtjt 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)rrrr rrarcsinrrs rasinr(Brrc|jtvrtdtjt 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)rrrr rrarcsinhrrs rasinhr+Nrrc|jtvrtdtjt 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)rrrr rrarctanrrs ratanr.Zrrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rrrr r r#rrarctan2rr%s ratan2r1fsx  x'''28;K+K+KIJJJ28$$$  &r2 . .FB :bjBI66 7 77rc|jtvrtdtjt 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)rrrr rrarctanhrrs ratanhr4urrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rrrr r r#rr bitwise_andrr%s rr6r6y 222 85 5 5STTT28$$$  &r2 . .FB :bnRY :: ; ;;rcv|jtvs|jtvrtdt|j|jt j||\}}t j|jdkrtdt j t 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) rrrr r r#ranyr ValueErrorr left_shiftr%s rbitwise_left_shiftr<s  x&&"(/*I*IOPPP28$$$  &r2 . .FB vbi!mSQRRR :bmBIry99 : ::rc|jtvrtdtjt 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) rrrr r r#rr bitwise_orrr%s rrArAsy 222 85 5 5RSSS28$$$  &r2 . .FB :bmBIry99 : ::rcv|jtvs|jtvrtdt|j|jt j||\}}t j|jdkrtdt j t 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) rrrr r r#rr9rr:r right_shiftr%s rbitwise_right_shiftrDs  x&&"(/*I*IPQQQ28$$$  &r2 . .FB vbi!mTRSSS :bnRY :: ; ;;rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rrrr r r#rr bitwise_xorrr%s rrFrFr7rc|jtvrtd|jtvr|St jt j|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) rr rrr rrceilrrs rrHrHsN  wo%%ABBBw/!! :bgah'' ( ((rc|jtvrtdtjt 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)rrrr rrcosrrs rrJrJ=  w&&&GHHH :bfQX&& ' ''rc|jtvrtdtjt 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)rrrr rrcoshrrs rrMrM=  w&&&HIII :bgah'' ( ((rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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 r#rrdividerr%s rrPrPsx  x'''28;K+K+KJKKK28$$$  &r2 . .FB :bi 2955 6 66rct|j|jtj||\}}tjt j|j|jS)zz Array API compatible wrapper for :py:func:`np.equal `. See its docstring for more information. )r rr r#rrequalrr%s rrRrRsN28$$$  &r2 . .FB :bhry")44 5 55rc|jtvrtdtjt 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)rrrr rrexprrs rrTrT!rKrc|jtvrtdtjt 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)rrrr rrexpm1rrs rrVrV,=  w&&&IJJJ :bhqx(( ) ))rc|jtvrtd|jtvr|St jt j|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) rr rrr rrfloorrrs rrYrY7N  wo%%BCCCw/!! :bhqx(( ) ))rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rr rr r r#rr floor_dividerr%s rr\r\Esv  x&&"(/*I*IIJJJ28$$$  &r2 . .FB :bobi;; < <`. See its docstring for more information. z*Only numeric dtypes are allowed in greater) rr rr r r#rrgreaterrr%s rr^r^Ssv  x&&"(/*I*IDEEE28$$$  &r2 . .FB :bjBI66 7 77rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rr rr r r#rr greater_equalrr%s rr`r`asw  x&&"(/*I*IJKKK28$$$  &r2 . .FB :b&ry")<< = ==rc|jtvrtdtjt 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 rr rrisfiniterrs rrbrbo<  wo%%EFFF :bk!(++ , ,,rc|jtvrtdtjt 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 rr rrisinfrrs rrerez<  wo%%BCCC :bhqx(( ) ))rc|jtvrtdtjt 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 rr rrisnanrrs rrhrhrfrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rr rr r r#rrlessrr%s rrjrjsv  x&&"(/*I*IABBB28$$$  &r2 . .FB :bgbi33 4 44rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rr rr r r#rr less_equalrr%s rrlrlv  x&&"(/*I*IGHHH28$$$  &r2 . .FB :bmBIry99 : ::rc|jtvrtdtjt 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)rrrr rrlogrrs rrororKrc|jtvrtdtjt 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)rrrr rrlog1prrs rrqrqrWrc|jtvrtdtjt 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)rrrr rrlog2rrs rrsrsrNrc|jtvrtdtjt 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)rrrr rrlog10rrs rrururWrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rrrr r r#rr logaddexprr%s rrwrwsx  x'''28;K+K+KMNNN28$$$  &r2 . .FB :bl29bi88 9 99rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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 r#rr logical_andrr%s rryryv  x&&"(/*I*IHIII28$$$  &r2 . .FB :bnRY :: ; ;;rc|jtvrtdtjt 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)rrrr rr logical_notrrs rr|r|s<  wo%%HIII :bnQX.. / //rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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 r#rr logical_orrr%s rr~r~rmrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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 r#rr logical_xorrr%s rrr rzrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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 r#rrmultiplyrr%s rrrv  x&&"(/*I*IEFFF28$$$  &r2 . .FB :bk")RY77 8 88rc|jtvrtdtjt 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 rr rrnegativerrs rrr)rcrct|j|jtj||\}}tjt j|j|jS)z Array API compatible wrapper for :py:func:`np.not_equal `. See its docstring for more information. )r rr r#rr not_equalrr%s rrr4sN28$$$  &r2 . .FB :bl29bi88 9 99rc|jtvrtdtjt 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 rr rrpositiverrs rrr@rcrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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 r#rrpowerrr%s rpowrLsv  x&&"(/*I*I@AAA28$$$  &r2 . .FB :bhry")44 5 55rc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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) rr rr r r#rr remainderrr%s rrrZsv  x&&"(/*I*IFGGG28$$$  &r2 . .FB :bl29bi88 9 99rc|jtvrtdtjt 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 rr rrroundrrs rrrhrfrc|jtvrtdtjt 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 rr rrsignrrs rrrss<  wo%%ABBB :bgah'' ( ((rc|jtvrtdtjt 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)rrrr rrsinrrs rrr~rKrc|jtvrtdtjt 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)rrrr rrsinhrrs rrrrNrc|jtvrtdtjt 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 rr rrsquarerrs rrrs<  wo%%CDDD :bi)) * **rc|jtvrtdtjt 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)rrrr rrsqrtrrs rrrrNrc|jtvs|jtvrtdt|j|jt j||\}}t jtj|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 r#rrsubtractrr%s rrrrrc|jtvrtdtjt 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)rrrr rrtanrrs rrrrKrc|jtvrtdtjt 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)rrrr rrtanhrrs rrrrNrc|jtvrtd|jtvr|St jt j|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) rr rrr rrtruncrrs rrrrZr)r r r r )r r r!r r r )E __future__r_dtypesrrrrr r _array_objectr numpyrrrrr$r(r+r.r1r4r6r<r?rArDrFrHrJrMrPrRrTrVrYr\r^r`rbrerhrjrlrorqrsrurwryr|r~rrrrrrrrrrrrrrrrrrrrs""""""! ((((++++,,,, 4 4 4 4++++,,,,++++ 8 8 8 8,,,,<<<<$;;;;$++++;;;;$<<<<"<<<<" ) ) ) )(((()))) 7 7 7 7 6 6 6 6((((**** * * * * = = = = 8 8 8 8 > > > >----******** 5 5 5 5 ; ; ; ;((((****))))**** : : : : < < < <0000 ; ; ; ; < < < < 9 9 9 9---- : : : :---- 6 6 6 6 : : : :****))))(((())))++++)))) 9 9 9 9(((()))) * * * * * *r