Мое мышление: Axis = n, где n = 0, 1 и т. д. означает, что матрица свертывается (складывается) вдоль этой оси. Итак, в 2D-матрице, когда вы рушитесь вдоль 0 (строк), вы действительно работаете по одному столбцу за раз. Аналогично для матриц более высокого порядка.
Это не то же самое, что нормальная ссылка на измерение в матрице, где 0 -> строка и 1 -> столбец. Аналогично для других измерений в массиве размерности N.