Speed up Matrix multiplication

조회 수: 6 (최근 30일)
Dustin Williams
Dustin Williams 2020년 4월 20일
댓글: Dustin Williams 2020년 4월 20일
I want to do the following Matrix Multiplication. Problem - Example:
A = rand(3,3,1000000);
B = rand(3,1000000)
How to calculate C faster than with this For-loop? I tried parfor but its only slightly faster.
Furthermore i'd prefer not to install the Parallel Computing Toolbox or MTIMESX. Is it possible just with reshape/permute/bsxfun?
C = zeros(3,1,size(B,2));
for idx=1:size(B,2)
C(:,:,idx) = A(:,:,idx) * B(:,idx);
end
Thanks!
  댓글 수: 1
Michael Soskind
Michael Soskind 2020년 4월 20일
I see a marginal performance improvement by simply getting rid of the middle array index:
A = rand(3,3,1e6);
B = rand(3,1e6);
% original code
tic
C = zeros(3,1,size(B,2));
for idx=1:size(B,2)
C(:,:,idx) = A(:,:,idx) * B(:,idx);
end
toc
% modified code with one fewer array dimension
tic
C = zeros(3,size(B,2));
for idx=1:size(B,2)
C(:,idx) = A(:,:,idx) * B(:,idx);
end
toc
Not sure if that helps or hurts you, but I get about 0.3 sec faster execution time, on average (Goes from 2 sec to 1.7 sec).
Michael

댓글을 달려면 로그인하십시오.

채택된 답변

Matt J
Matt J 2020년 4월 20일
편집: Matt J 2020년 4월 20일
C=sum( A.*reshape(B,1,3,[]),2);
  댓글 수: 2
Matt J
Matt J 2020년 4월 20일
Or, on pre-R2016b versions,
C=sum( bsxfun( @times, A, reshape(B,1,3,[]) ) ,2)
Dustin Williams
Dustin Williams 2020년 4월 20일
Thanks!

댓글을 달려면 로그인하십시오.

추가 답변 (0개)

카테고리

Help CenterFile Exchange에서 Logical에 대해 자세히 알아보기

태그

제품

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by