Given that A is a sparse matrix, norm(A(i,:)) takes a very long time. Why and can one do better ?

조회 수: 4 (최근 30일)
I am computing norm(A(i,:)) within a loop, where A is a sparse matrix. This seems to be a bottleneck, is there anyway to perform this faster ? Thank you.

채택된 답변

John D'Errico
John D'Errico 2020년 8월 10일
You really don't want to use loops to do something like this.
>> A = sprand(10000,10000,0.001);
>> timeit(@() norm(A(1,:)))
ans =
0.000121650132
>> timeit(@() sqrt(sum(A.^2,2)))
ans =
0.000442972132
So, the time to compute the norm of just one of the rows was a significant fraction of the total time to compute the norm of all rows at once.

추가 답변 (2개)

Walter Roberson
Walter Roberson 2020년 8월 11일
편집: Walter Roberson 2020년 8월 11일
If you were going to do this repeatedly, then
%one time cost
At = A.';
%then
norm(At(:,1))
is over 100 times faster than
norm(A(1,:))
For John's suggestion of A = sprand(10000,10000,0.001); then the cost of transposing A is roughly the same as the cost of calculating 11 row norms of A.
... And of course, it might happen to be practical to work with the transposed version instead of the original, so in some cases there would be no computation cost, just the cost of changing the code.

James Tursa
James Tursa 2020년 8월 9일
편집: James Tursa 2020년 8월 9일
Will you eventually need all of the rows? E.g., do this once at the beginning outside the loop
n = sqrt(sum(A.^2,2))
And then just pick off the element you want each iteration.
Alternatively, you could do this calculation one row at a time in a mex routine in order to avoid the data copy associated with forming the A(i,:) row explicitly.

카테고리

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

태그

Community Treasure Hunt

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

Start Hunting!

Translated by