用于计算总和的代码矢量化

Question

我想矢量化总和的计算

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

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

有很多方法可以编写我的方法，但我想知道是否有一个聪明的方法只适合一两行代码。

Answer 1

您可以先连接，然后沿第 3 个维度求和：

[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