diff options
Diffstat (limited to 'numpy/_array_api/_manipulation_functions.py')
| -rw-r--r-- | numpy/_array_api/_manipulation_functions.py | 23 |
1 files changed, 9 insertions, 14 deletions
diff --git a/numpy/_array_api/_manipulation_functions.py b/numpy/_array_api/_manipulation_functions.py index 19e9c1cab..262c712f8 100644 --- a/numpy/_array_api/_manipulation_functions.py +++ b/numpy/_array_api/_manipulation_functions.py @@ -1,28 +1,23 @@ +import numpy as np + def concat(arrays, /, *, axis=0): # Note: the function name is different here - from .. import concatenate - return concatenate(arrays, axis=axis) + return np.concatenate(arrays, axis=axis) def expand_dims(x, axis, /): - from .. import expand_dims - return expand_dims(x, axis) + return np.expand_dims(x, axis) def flip(x, /, *, axis=None): - from .. import flip - return flip(x, axis=axis) + return np.flip(x, axis=axis) def reshape(x, shape, /): - from .. import reshape - return reshape(x, shape) + return np.reshape(x, shape) def roll(x, shift, /, *, axis=None): - from .. import roll - return roll(x, shift, axis=axis) + return np.roll(x, shift, axis=axis) def squeeze(x, /, *, axis=None): - from .. import squeeze - return squeeze(x, axis=axis) + return np.squeeze(x, axis=axis) def stack(arrays, /, *, axis=0): - from .. import stack - return stack(arrays, axis=axis) + return np.stack(arrays, axis=axis) |