revert comment

Author: lezcano

torch_np/_util.py

@@ -127,7 +127,7 @@ def apply_keepdims(tensor, axis, ndim):
 
 
 def axis_none_flatten(*tensors, axis=None):
-    """Flatten the arrays if axis is None else normalize the axis."""
+    """Flatten the arrays if axis is None."""
     if axis is None:
         tensors = tuple(ar.flatten() for ar in tensors)
         return tensors, 0