python - 如何将 n 维 numpy 数组切割成给定大小的块

Question

我想在 python 中采用 n 维输入数组的形状和预期块大小的形状，并将数组切片成这些块。但是该函数仅获取数组的形状和预期的块作为参数，并且我只需要返回形成的切片的索引。

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) )
]