MAINT: better error message for wrong axis arguments

Co-authored-by: Matthew Barber <[email protected]>

Author: ev-br

torch_np/_detail/_reductions.py

@@ -25,6 +25,8 @@ def wrapped(tensor, axis, *args, **kwds):
 
         if axis is not None:
             if not isinstance(axis, (list, tuple)):
+                if not isinstance(axis, typing.SupportsIndex):
+                    raise TypeError(f"{type(axis)=}, but should be a list/tuple or support operator.index()")
                 axis = (axis,)
             axis = _util.normalize_axis_tuple(axis, tensor.ndim)