Is there a better vectorization technique?

Question

Kalamaya 2012-8-29

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/46984-is-there-a-better-vectorization-technique

I am trying to see if there are other ways of coding this code sample more efficiently. Here, y is an 1xM matrix, (say, 1x1000), and z is an NxM matrix, (say, 5x1000).

mean(ones(N,1)*y.^3 .* z,2)

This code works fine, but I worry of N increases a lot, that the:

ones(N,1)*y.^3

might get too wasteful and make everything slow down.

Thoughts?

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

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

Answer 1

John D'Errico 2012-8-29

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/46984-is-there-a-better-vectorization-technique#answer_57420

在 MATLAB Online 中打开

Its not THAT terrible for a matrix that small. However, you can gain from the use of bsxfun in many instances. Here, the matrices are simply too small to really gain anything.

>> N = 5;M =1000;
>> y = rand(1,M);
>> z = rand(N,M);
>> mean(ones(N,1)*y.^3 .* z,2)
ans =
      0.12412
      0.11669
      0.12102
      0.11976
      0.12196
>> mean(bsxfun(@times,y.^3,z),2)
ans =
      0.12412
      0.11669
      0.12102
      0.11976
      0.12196
>> z*y.'.^3/M
ans =
      0.12412
      0.11669
      0.12102
      0.11976
      0.12196

As you can see, all three solutions return the same result. All are equally valid.

Now I'll compare the times required.

>> timeit(@() mean(ones(N,1)*y.^3 .* z,2))
ans =
   0.00023018
>> timeit(@() mean(bsxfun(@times,y.^3,z),2))
ans =
   0.00026829
>> timeit(@() z*y.'.^3/M)
ans =
   0.00016594

As I said, you don't gain much. In fact, bsxfun does not gain at all, and is a bit slower. But you can gain a bit, if you re-write the expression into the third form I've posed. Not much, but a bit.

2 个评论
显示无隐藏无

Kalamaya 2012-8-29

I thought about using bsxfun. How is the timing if the N was very large, say, 2000 or something like that? (timeit is your own function?)

Jan 2012-8-29

编辑：Jan 2012-8-29

FEX: timeit

请先登录，再进行评论。

Answer 2

Jan 2012-8-29

2
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/46984-is-there-a-better-vectorization-technique#answer_57426

编辑：Jan 2012-8-30

在 MATLAB Online 中打开

The power() operator is more expensive than the matrix multiplication. Therefore an explicit multiplication saves time:

M = 1000;
N = 5;
y = rand(1, M);
z = rand(N, M);
tic; for i=1:100, a = mean(ones(N,1) * y .^ 3 .* z, 2); end; toc
% Elapsed time is 0.036668 seconds.
tic; for i=1:100, a = z * y.' .^ 3 / M; end; toc
% Elapsed time is 0.026818 seconds.
tic; for i=1:100, a = z * (y .* y .* y)' / M; end; toc
% Elapsed time is 0.001327 seconds.