How can I vectorize this function with nested FOR loop?

조회 수: 1 (최근 30일)
abtin irani
abtin irani 2019년 7월 2일
편집: Matt J 2019년 7월 5일
I have two for loops.L is one matrix of random numbers between 0,1 with dimension 24*100000.I want to vectorize it but i can't. because current code is very slow and take a long time.please help me.
K=zeros(100000,1);
T=zeros(100000,1);
for i=1:100000
for j=1:100000
K(j,1)=exp(-4*norm(L(:,i)-L(:,j))^2/norm(L(:,i))^2);
end
T(i,:)=sum(K)-1;
end
  댓글 수: 2
Bob Thompson
Bob Thompson 2019년 7월 2일
(L(:,i)-L(:,j))
This is what is going to make it difficult to vectorize. I believe there is a command to calculate this for you, but I'm not sure what exactly it is. It might be easier to do some research for this specifically, rather than the vectorization.
Jan
Jan 2019년 7월 2일
편집: Jan 2019년 7월 2일
Why do you want to vectorize the code? There is no general benefit in doing this. Does the current code run too slow? Then an acceleration is the way to go. Vectorizing can improve the speed, but this is not in general.
To optimize the code, we need the chance to run it. Without meaningful input arguments, this is hard. So please provide L.
norm(x)^2 calculates an expensive square root only to square the result afterwards.
Do oyu have the parallel processing toolbox? A parfor might be very useful.

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

채택된 답변

Matt J
Matt J 2019년 7월 2일
편집: Matt J 2019년 7월 2일
Using mat2tiles
chunksize=10000;
Lc=mat2tiles(L.',[chunksize,24]);
normL=mat2tiles(vecnorm(L,2,1), [1, chunksize]);
N=numel(Lc);
Tc=cell(N);
for i=1:N
for j=1:N
E=pdist2(Lc{i},Lc{j})./normL{j};
Tc{i,j}=sum( exp(-4*E.^2) ,1);
end
end
T=sum( cell2mat(Tc) ,1).'-1;
  댓글 수: 18
abtin irani
abtin irani 2019년 7월 4일
so,what can i do if i don't use pdist2.is there any other solution instead of pdist2 that doesn't give this error? i want one solution for chunksize=1000000. thank you for your answers.
Matt J
Matt J 2019년 7월 4일
편집: Matt J 2019년 7월 5일
Well, maybe the question to ask is, why are you making the chunksize larger, when you toldu us that the code already worked when the chunksize was just 10,000? Why change/fix something that is not broken?

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

추가 답변 (1개)

Jan
Jan 2019년 7월 4일
n = 1000;
L = rand(24, n);
T = zeros(n, 1);
for i=1:n
K = exp(-4*sum((L(:,i) - L) .^ 2, 1) ./ sum(L(:,i).^2, 1));
T2(i) = sum(K, 2) - 1;
end
This is 100 times faster than the original version for n=1000.
With parfor instead of for a further acceleration is possible.
To my surprise the above code is faster than this, which omits the repeated squaring:
T = zeros(n,1);
L2 = L .^ 2;
for i=1:n
K = exp(-4*sum((L2(:,i) - 2*L(:,i).*L + L2), 1) ./ sum(L2(:,i), 1));
T(i) = sum(K, 2)-1;
end
  댓글 수: 2
abtin irani
abtin irani 2019년 7월 4일
excuse me i got this error. Error using .*
Matrix dimensions must agree.
Error in Untitled8 (line 33)
K=exp(-4*sum((L2(:,i) - 2*L(:,i).*L + L2), 1) ./ sum(L2(:,i), 1));
Jan
Jan 2019년 7월 4일
You use Matlab < R2016b. Then:
K = exp(-4*sum((L2(:,i) - bsxfun(@times, 2*L(:,i), L) + L2), 1) ...
./ sum(L2(:,i), 1));
But the first version seems to be faster.

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

카테고리

Help CenterFile Exchange에서 Startup and Shutdown에 대해 자세히 알아보기

제품


릴리스

R2015a

Community Treasure Hunt

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

Start Hunting!

Translated by