The flatten func impl (#1009)

Author: Alexander-Makaryev

dpnp/dpnp_array.py

@@ -407,7 +407,37 @@ def flat(self):
 
         return dpnp.flatiter(self)
 
- # 'flatten',
+    def flatten(self, order='C'):
+        """
+        Return a copy of the array collapsed into one dimension.
+
+        Parameters
+        ----------
+        order: {'C', 'F', 'A', 'K'}, optional
+            'C' means to flatten in row-major (C-style) order.
+            'F' means to flatten in column-major (Fortran- style) order.
+            'A' means to flatten in column-major order if a is Fortran contiguous in memory, row-major order otherwise.
+            'K' means to flatten a in the order the elements occur in memory. The default is 'C'.
+
+        Returns
+        -------
+        out: ndarray
+            A copy of the input array, flattened to one dimension.
+
+        See Also
+        --------
+        :obj:`dpnp.ravel`, :obj:`dpnp.flat`
+
+        """
+
+        new_arr = dpnp.ndarray(self.shape, self.dtype)
+
+        if self.size > 0:
+            dpctl.tensor._copy_utils.copy_from_usm_ndarray_to_usm_ndarray(new_arr._array_obj, self._array_obj)
+            new_arr._array_obj = dpctl.tensor.reshape(new_arr._array_obj, (self.size, ))
+
+        return new_arr
+
  # 'getfield',
  # 'imag',
  # 'item',

tests/skipped_tests_gpu.tbl

@@ -339,27 +339,6 @@ tests/test_dparray.py::test_astype[[]-complex-int32]
 tests/test_dparray.py::test_astype[[]-complex-bool]
 tests/test_dparray.py::test_astype[[]-complex-bool_]
 tests/test_dparray.py::test_astype[[]-complex-complex]
-tests/test_dparray.py::test_flatten[[-2, -1, 0, 1, 2]-float64]
-tests/test_dparray.py::test_flatten[[-2, -1, 0, 1, 2]-float32]
-tests/test_dparray.py::test_flatten[[-2, -1, 0, 1, 2]-int64]
-tests/test_dparray.py::test_flatten[[-2, -1, 0, 1, 2]-int32]
-tests/test_dparray.py::test_flatten[[-2, -1, 0, 1, 2]-bool]
-tests/test_dparray.py::test_flatten[[-2, -1, 0, 1, 2]-bool_]
-tests/test_dparray.py::test_flatten[[-2, -1, 0, 1, 2]-complex]
-tests/test_dparray.py::test_flatten[[[-2, -1], [1, 2]]-float64]
-tests/test_dparray.py::test_flatten[[[-2, -1], [1, 2]]-float32]
-tests/test_dparray.py::test_flatten[[[-2, -1], [1, 2]]-int64]
-tests/test_dparray.py::test_flatten[[[-2, -1], [1, 2]]-int32]
-tests/test_dparray.py::test_flatten[[[-2, -1], [1, 2]]-bool]
-tests/test_dparray.py::test_flatten[[[-2, -1], [1, 2]]-bool_]
-tests/test_dparray.py::test_flatten[[[-2, -1], [1, 2]]-complex]
-tests/test_dparray.py::test_flatten[[]-float64]
-tests/test_dparray.py::test_flatten[[]-float32]
-tests/test_dparray.py::test_flatten[[]-int64]
-tests/test_dparray.py::test_flatten[[]-int32]
-tests/test_dparray.py::test_flatten[[]-bool]
-tests/test_dparray.py::test_flatten[[]-bool_]
-tests/test_dparray.py::test_flatten[[]-complex]
 tests/test_fft.py::test_fft[float32]
 tests/test_fft.py::test_fft[float64]
 tests/test_fft.py::test_fft[int32]