如何将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) )
]