vectorizing or speeding looped code
조회 수: 7 (최근 30일)
이전 댓글 표시
i am trying to speed up some code with multiple functions and have found the one that takes the most time (it is all a converted fortran code). it runs a nested for loop 4 times (each with modified input). i've had some success elsewhere in the code eliminating slow points but this one just cant get right. any ideas? (in 2016b for compatability reasons)
for II= 1:N
XP(1)= 1.0D0;
for JJ=2:IORD1
XP(JJ)=XP(JJ-1)*double(A1(II));
end
for JJ= 1:IORD1
for KK= 1:IORD1
B(JJ,KK)=B(JJ,KK)+XP(JJ)*XP(KK);
end
C(JJ)=C(JJ)+XP(JJ)*A2(II);
end
end
댓글 수: 0
답변 (3개)
Turlough Hughes
2020년 9월 22일
This should help, though I've already made assumptions about the sizes of arrays. How many rows/columns are in each variable?
XP(1) = 1;
for II= 1:N
XP(2:end) = XP(1:end-1)*double(A1(II));
B = B + XP(:).*XP(:).';
C = C + XP*A2(II);
end
댓글 수: 2
Turlough Hughes
2020년 9월 22일
Did the code work?
Some questions: is XP a column vector/row vector? Does A2 have as many elements as XP? Also, does B contain IORD1 rows and columns? Similarly does C have IORD1 elements?
Paul Schenk
2020년 9월 22일
댓글 수: 2
Turlough Hughes
2020년 9월 23일
Vectorisation is usually faster but not always. Probably best to just provide a minimum working example (google it) of code that I (or others) can test and profile. Otherwise it really is just guess work.
참고 항목
카테고리
Help Center 및 File Exchange에서 Loops and Conditional Statements에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!