a = a.reshape(2, 3, 4)
colon = slice(None)
dimensions = [(list(range(dim)) + [colon]) for dim in a.shape]
for dim in itertools.product(*dimensions):
    if dim.count(colon) == 1:
        pass
Итерация по всем одномерным подмассивам многомерного массива