Could maybe simplify this, as `axis=tuple(...)` is currently the only alternative to a single integer.