ENH: add np.ptp and ndarray.ptp

Author: ev-br

autogen/numpy_api_dump.py

@@ -774,9 +774,6 @@ def product(*args, **kwargs):
     raise NotImplementedError
 
 
-def ptp(a, axis=None, out=None, keepdims=NoValue):
-    raise NotImplementedError
-
 
 def put(a, ind, v, mode="raise"):
     raise NotImplementedError

torch_np/_detail/_reductions.py

@@ -89,6 +89,11 @@ def min(tensor, axis=None, initial=NoValue, where=NoValue):
     return result
 
 
+def ptp(tensor, axis=None):
+    result = tensor.amax(axis) - tensor.amin(axis)
+    return result
+
+
 def sum(tensor, axis=None, dtype=None, initial=NoValue, where=NoValue):
     if initial is not NoValue or where is not NoValue:
         raise NotImplementedError

torch_np/_ndarray.py

@@ -282,6 +282,7 @@ def nonzero(self):
     all = emulate_out_arg(axis_keepdims_wrapper(_reductions.all))
     max = emulate_out_arg(axis_keepdims_wrapper(_reductions.max))
     min = emulate_out_arg(axis_keepdims_wrapper(_reductions.min))
+    ptp = emulate_out_arg(axis_keepdims_wrapper(_reductions.ptp))
 
     sum = emulate_out_arg(axis_keepdims_wrapper(dtype_to_torch(_reductions.sum)))
     prod = emulate_out_arg(axis_keepdims_wrapper(dtype_to_torch(_reductions.prod)))

torch_np/_wrapper.py

@@ -604,6 +604,11 @@ def amin(a, axis=None, out=None, keepdims=NoValue, initial=NoValue, where=NoValu
 min = amin
 
 
+def ptp(a, axis=None, out=None, keepdims=NoValue):
+    arr = asarray(a)
+    return arr.ptp(axis=axis, out=out, keepdims=keepdims)
+
+
 def all(a, axis=None, out=None, keepdims=NoValue, *, where=NoValue):
     arr = asarray(a)
     return arr.all(axis=axis, out=out, keepdims=keepdims, where=where)

torch_np/tests/numpy_tests/lib/test_function_base.py

@@ -593,7 +593,6 @@ def test_basic(self):
         assert_equal(np.amin(b, axis=1), [3.0, 4.0, 2.0])
 
 
-@pytest.mark.xfail(reason='TODO: implement')
 class TestPtp:
 
     def test_basic(self):