如何以Z字形方式重新排序/重塑N-D矩阵？

Question

我想以这种方式对张量细胞进行排序

注意，每个单元格不是单个值，而是一堆值（DCT组件）。因此，原始张量的尺寸不是2D。