Function reshape_arr

numpy/lib/_function_base_impl.py:3875–3879

(a)

            nkeep = len(keep)

            def reshape_arr(a):
                # move axis that should not be reduced to front
                a = np.moveaxis(a, keep, range(nkeep))
                # merge reduced axis
                return a.reshape(a.shape[:nkeep] + (-1,))

            a = reshape_arr(a)