Matlab的“ arrayfun”功能

Question

考虑以下函数，该函数将灰度图像（2D矩阵）作为输入：

function r = fun1(img)
r = sum(sum(img));

我正在考虑使用arrayfun处理一系列图像（3d矩阵），从而消除了for循环的需要：

arrayfun(@fun1, imgStack);

但是arrayfun尝试将arrayfun每个元素imgStack视为fun1的输入，上一个操作的结果也是3D矩阵。 如何让arrayfun知道我只想在imgStack的第3维上重复fun1 ？

另一个问题， arrayfun是否arrayfun调用fun1 ？

Answer 1

在这种情况下，您不需要arrayfun来执行计算，只需执行以下操作：

imgStack = rand( 10, 10, 4 ); % 4 10x10 images
r = sum( sum( imgStack, 1 ), 2 ); % sum along both dimensions 1 and 2

通常，许多MATLAB操作将一次对整个数组进行操作，这是避免循环的常用方法。

MATLAB的常规“ arrayfun ”不是并行的。 但是，对于GPUArrays （具有并行计算工具箱），存在并行版本的arrayfun 。

Answer 2

关于第一个问题：您可以为此尝试accumarray。 一个建议

function ds = applyfun_onfirstdim(arr, h_fun)

dimvec = size(arr);
indexarr = repmat( (1:dimvec(1))', [1, dimvec(2:end)] );

ds = accumarray(indexarr(:), arr(:), [], h_fun);

这将创建与输入“ arr”相同尺寸的辅助索引数组。 您要应用h_fun的每个分片都获得相同的索引号。 在此示例中，它是第一个。