如何用三层嵌套循环将n×m×4维度矩阵展平为一维矩阵？

核心逻辑 #

要满足index % 4 == i的要求，本质是把三维矩阵中**同一通道（i值相同）**的元素按顺序集中排列在一维数组的对应位置：i=0对应0、4、8…；i=1对应1、5、9…，以此类推。

推导过程 #

1. 先确定当前(x,y)位置对应的是三维矩阵里的第几个4元素组：每个x行包含m个y列元素，因此第x行第y列的组序号为x * m + y（从0开始计数）。
2. 每个组在一维数组中占据4个连续位置，起始索引为组序号 * 4。
3. 加上当前通道索引i，即可得到最终的一维数组索引。

最终计算公式 #

let index = (x * m + y) * 4 + i;

验证示例 #

假设n=2，m=2：

• x=0,y=0,i=0 → index=(0*2+0)*4+0=0 → 0%4=0，符合要求
• x=0,y=1,i=1 → index=(0*2+1)*4+1=5 →5%4=1，符合要求
• x=1,y=0,i=2 → index=(1*2+0)*4+2=10 →10%4=2，符合要求
• x=1,y=1,i=3 → index=(1*2+1)*4+3=15 →15%4=3，符合要求

内容的提问来源于stack exchange，提问作者Ecanyte