summaryrefslogtreecommitdiff
path: root/numpy/_array_api/_manipulation_functions.py
diff options
context:
space:
mode:
authorAaron Meurer <asmeurer@gmail.com>2021-01-12 12:34:59 -0700
committerAaron Meurer <asmeurer@gmail.com>2021-01-12 12:34:59 -0700
commit9578636259f86267c2253f4af2510ce1eeaf084c (patch)
treef140af373c657aa2d3d107dd6504910bc5a47702 /numpy/_array_api/_manipulation_functions.py
parentd9651020aa9e1f6211b920954a357ac45712938d (diff)
downloadnumpy-9578636259f86267c2253f4af2510ce1eeaf084c.tar.gz
Make the array_api submodules private, and remove __all__ from individual files
The specific submodule organization is an implementation detail and should not be used. Only the top-level numpy._array_api namespace should be used.
Diffstat (limited to 'numpy/_array_api/_manipulation_functions.py')
-rw-r--r--numpy/_array_api/_manipulation_functions.py28
1 files changed, 28 insertions, 0 deletions
diff --git a/numpy/_array_api/_manipulation_functions.py b/numpy/_array_api/_manipulation_functions.py
new file mode 100644
index 000000000..19e9c1cab
--- /dev/null
+++ b/numpy/_array_api/_manipulation_functions.py
@@ -0,0 +1,28 @@
+def concat(arrays, /, *, axis=0):
+ # Note: the function name is different here
+ from .. import concatenate
+ return concatenate(arrays, axis=axis)
+
+def expand_dims(x, axis, /):
+ from .. import expand_dims
+ return expand_dims(x, axis)
+
+def flip(x, /, *, axis=None):
+ from .. import flip
+ return flip(x, axis=axis)
+
+def reshape(x, shape, /):
+ from .. import reshape
+ return reshape(x, shape)
+
+def roll(x, shift, /, *, axis=None):
+ from .. import roll
+ return roll(x, shift, axis=axis)
+
+def squeeze(x, /, *, axis=None):
+ from .. import squeeze
+ return squeeze(x, axis=axis)
+
+def stack(arrays, /, *, axis=0):
+ from .. import stack
+ return stack(arrays, axis=axis)