Julia中如何将4D数组重塑为矩阵构成的矩阵？

你用reshape(A, nK, nK)报错是必然的——reshape要求新维度的总元素数和原数组完全一致。你的4D数组A总元素数是nK×nK×m×m，而(nK,nK)的数组总元素数只有nK×nK，两者差了m²倍，维度不匹配的错误自然会出现。

要实现「(nK,nK)矩阵，每个元素是m×m矩阵」的结构，你需要构造一个元素类型为子矩阵的数组，而不是单纯调整原数组的维度拆分方式，具体做法如下（以Julia为例）：

方法1：嵌套循环赋值 #

# 假设原数组已定义
A = zeros(nK, nK, m, m)
# 初始化目标数组，元素类型为m×m浮点数矩阵
result = Matrix{Matrix{Float64}}(undef, nK, nK)
# 逐个提取原数组的子矩阵赋值
for i in 1:nK, j in 1:nK
    result[i,j] = A[i,j,:,:]
end

方法2：数组推导式（更简洁） #

result = [A[i,j,:,:] for i in 1:nK, j in 1:nK]

这样得到的result就是你需要的(nK,nK)结构，每个位置上的元素都是独立的m×m矩阵。

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