How to vectorize this type of function?

Question

Oliver 2013-5-2

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/74420-how-to-vectorize-this-type-of-function

I have a function of the form:

f(a) = sum((-1)^n/factorial(a-n))

where the summation limits are n = 0 to a. If a is a scalar then I can just substitute 0:a for n in the expression. But if a is a vector, then the limits on the sum are different for every element of a. One way to compute it would be:

for i = 1:numel(a)
    n = 0:a(i);
    f(i) = sum((-1).^n./factorial(a(i)-n));
end

However, I would like to avoid using a for loop. I am wondering if anyone has any ideas for how to vectorize this?

Matt J provided a great solution to a simpler problem HERE. Is there any way to do something similar?

3 个评论
显示 1更早的评论隐藏 1更早的评论

Oliver 2013-5-2

In practice, I'm using exp(gammaln(a(i)-n+1)) to avoid overflow. Just for argument's sake, lets say 'a' is never larger than 32. This is a simplified version of the real sum that I'm interested in, which is similar in form, but has 6 arguments and factorials depending on all of them (i.e. f(a,b,c,d,e,f)).

Oliver 2013-5-2

Actually, more realistic is a < 10.

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Matt J 2013-5-2

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/74420-how-to-vectorize-this-type-of-function#answer_84243

在 MATLAB Online 中打开

   nn=0:max(a);
    T=[1,cumprod(1:nn(end))];
     T=cumsum((-1).^nn./T);
     f=T(a+1).*(-1).^mod(a,2),

3 个评论
显示 1更早的评论隐藏 1更早的评论

Oliver 2013-5-3

Nevermind, it was just confusion of row vs. column vectors. Works great now. So that I can improve my coding, how did you come up with this? It seems like with this solution and the other one that you recently posted, a lot of it is based on computing the worst case scenario and using cumsum, and then cleverly indexing into the right portions of the vector of cumulative sums. Is this in general how you try to solve these types of problems?

Matt J 2013-5-3

编辑：Matt J 2013-5-3

The general idea is to create a look-up table, T, of all the things you are trying to evaluate and then index into the table with look-up values like a, nzmin, nzmax. Look-up tables are things that can often be filled sequentially/incrementally, using efficient vectorized MATLAB commands like cumsum or cumprod. Indexing into these tables is, of course, also vectorizable.

The strategy might not be so optimal if your look-up values were not reasonably dense throughout the table or if the cost of creating the table was high. For that, you might make several sub-tables or fall back to a for-loop.

请先登录，再进行评论。