Matlab：沿3D数组的第一维进行操作

Question

我有一个3D数组M（d * d，m，n）。 对于M的每个d * d向量（即第一维的向量），我将其分成d个部分，并取每个部分的总和形成一个新的向量（大小为d）。 例如，如果u是沿着M的第一个维度的向量，则它将被向量v代替，该向量由以下公式计算：

v = sum(reshape(u,d,d))';

目前，我使用如下循环，但是我认为应该有一种更快的方法。

N = zeros(d,m,n)
for i=1:m
    for j=1:n
        N(:,i,j) = sum(reshape(M(:,i,j),d,d))'; %//'
    end
end

非常感谢您的任何建议！

Answer 1

尝试这个 -

N = reshape(sum(reshape(M,d,[])),d,m,n)

Answer 2

我可能没有正确理解问题，但这是您要寻找的吗？

N=squeeze(sum(reshape(M,[d,d,size(M,2),size(M,3)])))