Fast multiplication of rows of one matrix and columns of the second matrix

Question

Mateusz 2011-10-5

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/17511-fast-multiplication-of-rows-of-one-matrix-and-columns-of-the-second-matrix

I would like to compute v(k) = A(k, :)* B(:, k) as fast as possible (no-loops). Currently, I am doing diag(A * B) but it has unnecessary overhead of computation, and storage.

2 个评论
显示无隐藏无

Daniel Shub 2011-10-5

Do you want to do it as fast as possible or without loops? The JIT accelerator means that those two things are not necessarily the same.

Daniel Shub 2011-10-5

How big is A? Is it sparse or distributed or anything funky like that?

请先登录，再进行评论。

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

Answer 1

Teja Muppirala 2011-10-5

2
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/17511-fast-multiplication-of-rows-of-one-matrix-and-columns-of-the-second-matrix#answer_23599

在 MATLAB Online 中打开

The fastest way to do something generally depends on the size and structure of your data.

Don't assume loops are slower. For simple linear algebra, loops are generally very fast. In fact for large matrices (1000x1000 etc.), I think loops are probably the fastest way actually.

v = zeros(1,size(A,1));
for k = 1:size(A,1)
  v(k) = A(k,:)*B(:,k);
end

For smaller matrices, you are probably better off doing this:

v = sum(A'.*B);

The best thing to do it just to try things out and see what works best for your data.

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

Teja Muppirala 2011-10-5

Ah. Yeah I forgot the dot. Thanks James

Dr. Seis 2011-10-5

I created two random 10000x10000 matrices and the "for loop" took 2 seconds to compute what "diag" took over 20 seconds to compute. However, and I will have to make a correction to the above, this took only 1 second to execute:

v = sum(A.*B',2);

Note: I added the "dot" to denote that each element in A is multiplied to each respective element of B-transpose before the rows are summed. This should be the same result as v = diag(A*B);

请先登录，再进行评论。

Answer 2

Daniel Shub 2011-10-5

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/17511-fast-multiplication-of-rows-of-one-matrix-and-columns-of-the-second-matrix#answer_23620

Yair's blog has a nice post on memory issues and array operations:

http://undocumentedmatlab.com/blog/matrix-processing-performance/

It is not always obvious what is the best solution.

If your matrices are really big you might be better off distributing them to a graphic card. If your machine has a lot of cores, the for loop could be replaced by a parfor loop, or even distributed to a cluster. It is silly to worry about slight inefficiencies if you can access 1000+ cores.