Speed up Matrix multiplication
조회 수: 6 (최근 30일)
이전 댓글 표시
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
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
채택된 답변
추가 답변 (0개)
참고 항목
카테고리
Help Center 및 File Exchange에서 Logical에 대해 자세히 알아보기
제품
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!