matlab - arrayfun 当数组的每一行都是输入时

Question

我的情况是，我想通过句柄为 fun 的函数映射一个标量数组 A，将一个行向量发送到一个行向量，以获取 B，这样 B(i,:) = fun(A(i,:))。

我能想到的最合理的解决方案如下:

temp = mat2cell(A,ones(1,size(A,1)),size(A,2));
B = cell2mat(cellfun(fun,temp,'UniformOutput',0));

但是，转换为单元格和返回单元格似乎有点矫枉过正(并且可能计算量大)。我也不清楚为什么 cellfun 提示输出不均匀。想到更有效的方法了吗？

Answer 1

还有另一种使用 accumarray 的解决方案.虽然不像 bsxfun 那样花哨，但它不需要声明辅助函数:

subs = ndgrid(1:size(A, 1));
B = accumarray(subs(:), A(:), [], @fun); %// @fun is the function handle