Use positional/keyword only args delimiter in manipulation functions

Author: oleksandr-pavlyk

dpctl/tensor/_manipulation_functions.py

@@ -83,7 +83,7 @@ def _broadcast_shapes(*args):
     return _broadcast_shape_impl(array_shapes)
 
 
-def permute_dims(X, axes):
+def permute_dims(X, /, axes):
     """permute_dims(x, axes)
 
     Permute the axes (dimensions) of an array; returns the permuted
@@ -120,7 +120,7 @@ def permute_dims(X, axes):
     )
 
 
-def expand_dims(X, axis):
+def expand_dims(X, /, *, axis=0):
     """expand_dims(x, axis)
 
     Expands the shape of an array by inserting a new axis (dimension)
@@ -166,7 +166,7 @@ def expand_dims(X, axis):
     return dpt.reshape(X, shape)
 
 
-def squeeze(X, axis=None):
+def squeeze(X, /, axis=None):
     """squeeze(x, axis)
 
     Removes singleton dimensions (axes) from array `x`.
@@ -211,7 +211,7 @@ def squeeze(X, axis=None):
         return dpt.reshape(X, new_shape)
 
 
-def broadcast_to(X, shape):
+def broadcast_to(X, /, shape):
     """broadcast_to(x, shape)
 
     Broadcast an array to a new `shape`; returns the broadcasted
@@ -277,7 +277,7 @@ def broadcast_arrays(*args):
     return [broadcast_to(X, shape) for X in args]
 
 
-def flip(X, axis=None):
+def flip(X, /, *, axis=None):
     """flip(x, axis)
 
     Reverses the order of elements in an array `x` along the given `axis`.
@@ -309,7 +309,7 @@ def flip(X, axis=None):
     return X[indexer]
 
 
-def roll(X, shift, axis=None):
+def roll(X, /, shift, *, axis=None):
     """
     roll(x, shift, axis)
 
@@ -467,7 +467,7 @@ def _concat_axis_None(arrays):
     return res
 
 
-def concat(arrays, axis=0):
+def concat(arrays, /, *, axis=0):
     """concat(arrays, axis)
 
     Joins a sequence of arrays along an existing axis.
@@ -535,7 +535,7 @@ def concat(arrays, axis=0):
     return res
 
 
-def stack(arrays, axis=0):
+def stack(arrays, /, *, axis=0):
     """
     stack(arrays, axis)
 
@@ -596,7 +596,7 @@ def stack(arrays, axis=0):
     return res
 
 
-def unstack(X, axis=0):
+def unstack(X, /, *, axis=0):
     """unstack(x, axis=0)
 
     Splits an array in a sequence of arrays along the given axis.
@@ -625,7 +625,7 @@ def unstack(X, axis=0):
     return tuple(Y[i] for i in range(Y.shape[0]))
 
 
-def moveaxis(X, source, destination):
+def moveaxis(X, source, destination, /):
     """moveaxis(x, source, destination)
 
     Moves axes of an array to new positions.
@@ -714,7 +714,7 @@ def swapaxes(X, axis1, axis2):
     return dpt.permute_dims(X, tuple(ind))
 
 
-def repeat(x, repeats, axis=None):
+def repeat(x, repeats, /, *, axis=None):
     """repeat(x, repeats, axis=None)
 
     Repeat elements of an array.
@@ -930,7 +930,7 @@ def repeat(x, repeats, axis=None):
     return res
 
 
-def tile(x, repetitions):
+def tile(x, repetitions, /):
     """tile(x, repetitions)
 
     Repeat an input array `x` along each axis a number of times given by

dpctl/tensor/_reshape.py

@@ -87,7 +87,7 @@ def reshaped_strides(old_sh, old_sts, new_sh, order="C"):
     return new_sts if valid else None
 
 
-def reshape(X, shape, order="C", copy=None):
+def reshape(X, /, shape, *, order="C", copy=None):
     """reshape(x, shape, order="C")
 
     Reshapes array `x` into new shape.