Объединить ndarray вдоль одной оси

Question

Как я могу объединить массив на заданной оси? Так для следующего массива:

a = np.array([[[1,2],[3,4]], [[4,5],[6,7]]])

Я пытаюсь получить:

array([[1, 2, 3, 4],
       [4, 5, 6, 7]])   

Я пытался использовать np.concatenate, указав ось, но без удачи.