Add out keyword to docstring and update tests

Author: vlad-perevezentsev

dpctl/tensor/_elementwise_funcs.py

@@ -481,7 +481,7 @@
 
 # B20: ==== NOT_EQUAL   (x1, x2)
 _not_equal_docstring_ = """
-not_equal(x1, x2, order='K')
+not_equal(x1, x2, out=None, order='K')
 
 Calculates inequality test results for each element `x1_i` of the
 input array `x1` with the respective element `x2_i` of the input array `x2`.
@@ -491,6 +491,12 @@
         First input array, expected to have numeric data type.
     x2 (usm_ndarray):
         Second input array, also expected to have numeric data type.
+    out ({None, usm_ndarray}, optional):
+        Output array to populate.
+        Array have the correct shape and the expected data type.
+    order ("C","F","A","K", optional):
+        Memory layout of the newly output array, if parameter `out` is `None`.
+        Default: "K".
 Returns:
     usm_narray:
         an array containing the result of element-wise inequality comparison.

dpctl/tests/elementwise/test_equal.py

@@ -127,6 +127,10 @@ def test_equal_broadcasting():
     r2 = dpt.equal(v, m)
     assert (dpt.asnumpy(r2) == expected).all()
 
+    r3 = dpt.empty_like(m, dtype="?")
+    dpt.equal(m, v, r3)
+    assert (dpt.asnumpy(r3) == expected).all()
+
 
 @pytest.mark.parametrize("arr_dt", _all_dtypes)
 def test_equal_python_scalar(arr_dt):

dpctl/tests/elementwise/test_not_equal.py

@@ -127,6 +127,10 @@ def test_not_equal_broadcasting():
     r2 = dpt.not_equal(v, m)
     assert (dpt.asnumpy(r2) == expected).all()
 
+    r3 = dpt.empty_like(m, dtype="?")
+    dpt.not_equal(m, v, r3)
+    assert (dpt.asnumpy(r3) == expected).all()
+
 
 @pytest.mark.parametrize("arr_dt", _all_dtypes)
 def test_not_equal_python_scalar(arr_dt):