diff options
| author | Aaron Meurer <asmeurer@gmail.com> | 2021-01-11 16:36:11 -0700 |
|---|---|---|
| committer | Aaron Meurer <asmeurer@gmail.com> | 2021-01-11 16:36:11 -0700 |
| commit | 9934cf3abcd6ba9438c340042e94f8343e3f3d13 (patch) | |
| tree | fb8769bc4158180207d72f8028a8c753f0fed8aa /numpy/_array_api | |
| parent | 012343dec5599418b77512733fc5b8db6bc14c4c (diff) | |
| download | numpy-9934cf3abcd6ba9438c340042e94f8343e3f3d13.tar.gz | |
Add dtypes to the _array_api namespace
Diffstat (limited to 'numpy/_array_api')
| -rw-r--r-- | numpy/_array_api/__init__.py | 4 | ||||
| -rw-r--r-- | numpy/_array_api/dtypes.py | 3 |
2 files changed, 7 insertions, 0 deletions
diff --git a/numpy/_array_api/__init__.py b/numpy/_array_api/__init__.py index 878251e7c..1677224c5 100644 --- a/numpy/_array_api/__init__.py +++ b/numpy/_array_api/__init__.py @@ -8,6 +8,10 @@ from .creation_functions import arange, empty, empty_like, eye, full, full_like, __all__ += ['arange', 'empty', 'empty_like', 'eye', 'full', 'full_like', 'linspace', 'ones', 'ones_like', 'zeros', 'zeros_like'] +from .dtypes import int8, int16, int32, int64, uint8, uint16, uint32, uint64, float32, float64, bool + +__all__ += ['int8', 'int16', 'int32', 'int64', 'uint8', 'uint16', 'uint32', 'uint64', 'float32', 'float64', 'bool'] + from .elementwise_functions import 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 __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'] diff --git a/numpy/_array_api/dtypes.py b/numpy/_array_api/dtypes.py new file mode 100644 index 000000000..62fb3d321 --- /dev/null +++ b/numpy/_array_api/dtypes.py @@ -0,0 +1,3 @@ +from .. import int8, int16, int32, int64, uint8, uint16, uint32, uint64, float32, float64, bool + +__all__ = ['int8', 'int16', 'int32', 'int64', 'uint8', 'uint16', 'uint32', 'uint64', 'float32', 'float64', 'bool'] |