diff options
| author | Aaron Meurer <asmeurer@gmail.com> | 2021-01-11 16:03:43 -0700 |
|---|---|---|
| committer | Aaron Meurer <asmeurer@gmail.com> | 2021-01-11 16:03:43 -0700 |
| commit | 012343dec5599418b77512733fc5b8db6bc14c4c (patch) | |
| tree | 13ef3a6bfdcd603036baaf8d65d1647eea749afc /numpy/_array_api/elementwise_functions.py | |
| parent | 33dc7bea24f1ab6c47047b49521e732caeb485d5 (diff) | |
| download | numpy-012343dec5599418b77512733fc5b8db6bc14c4c.tar.gz | |
Add initial array_api sub-namespace
This is based on the function stubs from the array API test suite, and is
currently based on the assumption that NumPy already follows the array API
standard. Now it needs to be modified to fix it in the places where NumPy
deviates (for example, different function names for inverse trigonometric
functions).
Diffstat (limited to 'numpy/_array_api/elementwise_functions.py')
| -rw-r--r-- | numpy/_array_api/elementwise_functions.py | 221 |
1 files changed, 221 insertions, 0 deletions
diff --git a/numpy/_array_api/elementwise_functions.py b/numpy/_array_api/elementwise_functions.py new file mode 100644 index 000000000..3e8349d29 --- /dev/null +++ b/numpy/_array_api/elementwise_functions.py @@ -0,0 +1,221 @@ +def abs(x, /): + from .. import abs + return abs(x) + +def acos(x, /): + from .. import acos + return acos(x) + +def acosh(x, /): + from .. import acosh + return acosh(x) + +def add(x1, x2, /): + from .. import add + return add(x1, x2) + +def asin(x, /): + from .. import asin + return asin(x) + +def asinh(x, /): + from .. import asinh + return asinh(x) + +def atan(x, /): + from .. import atan + return atan(x) + +def atan2(x1, x2, /): + from .. import atan2 + return atan2(x1, x2) + +def atanh(x, /): + from .. import atanh + return atanh(x) + +def bitwise_and(x1, x2, /): + from .. import bitwise_and + return bitwise_and(x1, x2) + +def bitwise_left_shift(x1, x2, /): + from .. import bitwise_left_shift + return bitwise_left_shift(x1, x2) + +def bitwise_invert(x, /): + from .. import bitwise_invert + return bitwise_invert(x) + +def bitwise_or(x1, x2, /): + from .. import bitwise_or + return bitwise_or(x1, x2) + +def bitwise_right_shift(x1, x2, /): + from .. import bitwise_right_shift + return bitwise_right_shift(x1, x2) + +def bitwise_xor(x1, x2, /): + from .. import bitwise_xor + return bitwise_xor(x1, x2) + +def ceil(x, /): + from .. import ceil + return ceil(x) + +def cos(x, /): + from .. import cos + return cos(x) + +def cosh(x, /): + from .. import cosh + return cosh(x) + +def divide(x1, x2, /): + from .. import divide + return divide(x1, x2) + +def equal(x1, x2, /): + from .. import equal + return equal(x1, x2) + +def exp(x, /): + from .. import exp + return exp(x) + +def expm1(x, /): + from .. import expm1 + return expm1(x) + +def floor(x, /): + from .. import floor + return floor(x) + +def floor_divide(x1, x2, /): + from .. import floor_divide + return floor_divide(x1, x2) + +def greater(x1, x2, /): + from .. import greater + return greater(x1, x2) + +def greater_equal(x1, x2, /): + from .. import greater_equal + return greater_equal(x1, x2) + +def isfinite(x, /): + from .. import isfinite + return isfinite(x) + +def isinf(x, /): + from .. import isinf + return isinf(x) + +def isnan(x, /): + from .. import isnan + return isnan(x) + +def less(x1, x2, /): + from .. import less + return less(x1, x2) + +def less_equal(x1, x2, /): + from .. import less_equal + return less_equal(x1, x2) + +def log(x, /): + from .. import log + return log(x) + +def log1p(x, /): + from .. import log1p + return log1p(x) + +def log2(x, /): + from .. import log2 + return log2(x) + +def log10(x, /): + from .. import log10 + return log10(x) + +def logical_and(x1, x2, /): + from .. import logical_and + return logical_and(x1, x2) + +def logical_not(x, /): + from .. import logical_not + return logical_not(x) + +def logical_or(x1, x2, /): + from .. import logical_or + return logical_or(x1, x2) + +def logical_xor(x1, x2, /): + from .. import logical_xor + return logical_xor(x1, x2) + +def multiply(x1, x2, /): + from .. import multiply + return multiply(x1, x2) + +def negative(x, /): + from .. import negative + return negative(x) + +def not_equal(x1, x2, /): + from .. import not_equal + return not_equal(x1, x2) + +def positive(x, /): + from .. import positive + return positive(x) + +def pow(x1, x2, /): + from .. import pow + return pow(x1, x2) + +def remainder(x1, x2, /): + from .. import remainder + return remainder(x1, x2) + +def round(x, /): + from .. import round + return round(x) + +def sign(x, /): + from .. import sign + return sign(x) + +def sin(x, /): + from .. import sin + return sin(x) + +def sinh(x, /): + from .. import sinh + return sinh(x) + +def square(x, /): + from .. import square + return square(x) + +def sqrt(x, /): + from .. import sqrt + return sqrt(x) + +def subtract(x1, x2, /): + from .. import subtract + return subtract(x1, x2) + +def tan(x, /): + from .. import tan + return tan(x) + +def tanh(x, /): + from .. import tanh + return tanh(x) + +def trunc(x, /): + from .. import trunc + return trunc(x) + +__all__ = ['abs', 'acos', 'acosh', 'add', 'asin', 'asinh', 'atan', 'atan2', 'atanh', 'bitwise_and', 'bitwise_left_shift', 'bitwise_invert', 'bitwise_or', 'bitwise_right_shift', 'bitwise_xor', 'ceil', 'cos', 'cosh', 'divide', 'equal', 'exp', 'expm1', 'floor', 'floor_divide', 'greater', 'greater_equal', 'isfinite', 'isinf', 'isnan', 'less', 'less_equal', 'log', 'log1p', 'log2', 'log10', 'logical_and', 'logical_not', 'logical_or', 'logical_xor', 'multiply', 'negative', 'not_equal', 'positive', 'pow', 'remainder', 'round', 'sign', 'sin', 'sinh', 'square', 'sqrt', 'subtract', 'tan', 'tanh', 'trunc'] |
