Align real and imag signatures with ones from dpctl

Author: antonwolfy

dpnp/dpnp_algo/dpnp_elementwise_common.py

@@ -498,6 +498,27 @@ def __call__(self, x, deg=False):
         return res
 
 
+class DPNPImag(DPNPUnaryFunc):
+    """Class that implements dpnp.imag unary element-wise functions."""
+
+    def __init__(
+        self,
+        name,
+        result_type_resolver_fn,
+        unary_dp_impl_fn,
+        docs,
+    ):
+        super().__init__(
+            name,
+            result_type_resolver_fn,
+            unary_dp_impl_fn,
+            docs,
+        )
+
+    def __call__(self, x, out, order):
+        return super().__call__(x, out=out, order=order)
+
+
 class DPNPReal(DPNPUnaryFunc):
     """Class that implements dpnp.real unary element-wise functions."""
 
@@ -515,9 +536,9 @@ def __init__(
             docs,
         )
 
-    def __call__(self, x):
+    def __call__(self, x, out, order):
         if numpy.iscomplexobj(x):
-            return super().__call__(x)
+            return super().__call__(x, out=out, order=order)
         return x
 
 

dpnp/dpnp_iface_mathematical.py

@@ -62,6 +62,7 @@
 from .dpnp_algo.dpnp_elementwise_common import (
     DPNPAngle,
     DPNPBinaryFunc,
+    DPNPImag,
     DPNPReal,
     DPNPRound,
     DPNPUnaryFunc,
@@ -2355,12 +2356,6 @@ def gradient(f, *varargs, axis=None, edge_order=1):
     data type, the returned array has a floating-point data type
     with the same floating-point precision as complex input.
 
-Limitations
------------
-Parameters `where' and `subok` are supported with their default values.
-Keyword argument `kwargs` is currently unsupported.
-Otherwise ``NotImplementedError`` exception will be raised.
-
 See Also
 --------
 :obj:`dpnp.real` : Return the real part of the complex argument.
@@ -2385,7 +2380,7 @@ def gradient(f, *varargs, axis=None, edge_order=1):
 array(1.)
 """
 
-imag = DPNPUnaryFunc(
+imag = DPNPImag(
     "imag",
     ti._imag_result_type,
     ti._imag,
@@ -3331,6 +3326,13 @@ def prod(
 ----------
 x : {dpnp.ndarray, usm_ndarray}
     Input array, expected to have numeric data type.
+out : {None, dpnp.ndarray, usm_ndarray}, optional
+    Output array to populate.
+    Array must have the correct shape and the expected data type.
+    Default: ``None``.
+order : {"C", "F", "A", "K"}, optional
+    Memory layout of the newly output array, if parameter `out` is ``None``.
+    Default: ``"K"``.
 
 Returns
 -------