3d массив Numpy до 2d

У меня есть 3d-матрица, подобная этой

arange(16).reshape((4,2,2))
array([[[ 0,  1],
        [ 2,  3]],

        [[ 4,  5],
        [ 6,  7]],

        [[ 8,  9],
        [10, 11]],

        [[12, 13],
        [14, 15]]])

и хотел бы сгруппировать их в формате сетки, в результате получив

array([[ 0,  1,  4,  5],
       [ 2,  3,  6,  7],
       [ 8,  9, 12, 13],
       [10, 11, 14, 15]])

Есть ли способ сделать без явного hstacking (и/или vstacking) их или добавить дополнительное измерение и изменение формы (не уверен, что это сработает)?

Спасибо,