python - 将 3D 数字数组拆分为多个小数组

Question

我想将一个图像数组（带宽 x 宽度 x 高度）拆分为 N 个较小的数组，并将这些小数组存储在一个 numpy 数组中，如（N x 带宽 x 宽度/N x 高度/N）

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

标签： pythonnumpy