The range of valid values for the `axis` argument should be `[-rank(input), rank(input))`, just like other reduction ops.
PiperOrigin-RevId:
191657767
input_tensor: The tensor to reduce. Should have numeric type.
axis: The dimensions to reduce. If `None` (the default),
reduces all dimensions. Must be in the range
- `[-rank(input_tensor), rank(input_tensor)]`.
+ `[-rank(input_tensor), rank(input_tensor))`.
keepdims: If true, retains reduced dimensions with length 1.
name: A name for the operation (optional).
reduction_indices: The old (deprecated) name for axis.