Как нарезать n-мерный массив NumPy в блоки данного размера

Question

Я хочу взять форму n-мерного входного массива и форму ожидаемого размера блока в python и разрезать массив на эти блоки. Но функция получает в качестве аргументов только форму массива и ожидаемый блок, и мне нужно возвращать только индексы сформированных срезов.

def foo(input_shape, block_shape):
     **dome something**
     return indices of blocks formed with shape= block_shape

Например:

>>> x = foo((2,2,3), (2,2,2))
>>> print(x) 
[( slice(0,2), slice(0,2), slice(0,2) ), 
   (slice(0,2), slice(0,2), slice(1,3) )
]