specify axis in _range_check