use keepdims in docstring