代码的矢量化以计算总和

Question

我想对和的计算进行矢量化

A{1} + A{2} + ... + A{end}

其中A是一个矢量单元数组，每个A {i}是一个nxn数字矩阵。

有很多方法可以通过这种方式编写代码，但我想知道是否有一个聪明的方法只适用于一行或两行代码。

Answer 1

你可以先安排，然后按照第三维进行总结：

[A{1:3}] = deal(ones(4)); %# each element of the cell array contains a 4-by-4 array of ones

out = sum( cat(3,A{:}), 3); %# catenate, then sum