我目前正在处理具有二维矩阵/维度数组(n,pxq)的数据集。我必须 reshape 这些矩阵以获得大小为 (n,q,p) 的 3D 数组。
目前，我陷入困境
# Assuming mat2 exists
mat3 = zeros(n,q,p)
for r in 1:n
mat3[r,:,:] = transpose(reshape(mat2[r,:],p,q))
end
有没有办法避免这种操作，或者至少对其进行一些优化？
最佳答案
使用reshape + permutedims :
mat3 = permutedims(reshape(mat2, n, p, q), (1, 3, 2))
关于julia - 有效地 reshape 数组 2D -> 3D，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/66320554/