Update torch_np/_detail/_util.py

Fuse logical ops in normalize_axis_index

Co-authored-by: Mario Lezcano Casado <[email protected]>

Author: ev-br

torch_np/_detail/_util.py

@@ -35,7 +35,7 @@ class UFuncTypeError(TypeError, RuntimeError):
 
 # a replica of the version in ./numpy/numpy/core/src/multiarray/common.h
 def normalize_axis_index(ax, ndim, argname=None):
-    if ax < -ndim or ax >= ndim:
+    if not (-ndim <= ax < ndim):
         raise AxisError(f"axis {ax} is out of bounds for array of dimension {ndim}")
     if ax < 0:
         ax += ndim