将3D数字阵列拆分为多个小数组

Question

我想将图像数组（带x宽x高）分成N个较小的数组，并将小数组存储在一个numpy数组中（N x band x width / N x height / N）

例如，如果band = 4，width = 16且height = 16，则输入数组的形状为4x16x16。比方说，N = 4然后我想将我的输入数组分成四个小数组并将其存储在一个形状为4x4x4x4的输出数组中。如果输入数组的形状不能被N整除，那么我想首先重塑输入数组，使其宽度和高度可被N整除，然后进行拆分。