in the kwargs, the argument axis=None may be give when reducing all
axis. Handle this case.