Get the diagonal without calculating the explicit matrix

1 次查看（过去 30 天）
Long Hong
Long Hong 2020-3-26
编辑： Matt J 2020-3-26
Dear all:
I am trying to calculate a diagonal of a matrix (denoted A), which is formed by multiplying two large-dimensional matrices (denoted as B*C).
A naive way to do it is: first, calculating explicitly A = B*C, then get diagonal out from A. However, the first step takes forever to run due to the high-dimension of B and C. But the only thing I need is the diagonal of A.
Another straightforward way in my mind is: I could create a loop by calculating each element of the diagonal of A one by one. It will surely save a lot of time, but I am not sure if this is the most efficient way.
I am wondering if anyone knows a faster/smarter way to calculate it.
Thank you very much in advance!
Best,
Long
  1 个评论
Matt J
Matt J 2020-3-26
编辑：Matt J 2020-3-26
The best approach will depend on the dimensions of the matrices, and whether they are of sparse-type or not.

请先登录，再进行评论。

请先登录，再回答此问题。

采纳的回答

Matt J
Matt J 2020-3-26
编辑：Matt J 2020-3-26
Assuming B*C results in a square matrix,
diagonal=sum(B.' .* C, 1);
  8 个评论
显示 6更早的评论
Long Hong
Long Hong 2020-3-26
Thank you the cyclist! Do you have any insight in doing the transpose quicker? I am a bit confused here.
Matt J
Matt J 2020-3-26
编辑：Matt J 2020-3-26
the cyclist means you might avoid the transpose by loading data column-wise instead of row-wise when you first build B.

请先登录，再进行评论。

更多回答（1 个）

the cyclist
the cyclist 2020-3-26
Here is one way:
% Make up some inputs
N = 4;
B = rand(N);
C = rand(N);
% Calculate the diagonal
A_diag = 0;
for nr = 1:N
A_diag = A_diag + B(:,nr).*C(nr,:)';
end

类别

在 Help Center 和 File Exchange 中查找有关 Operating on Diagonal Matrices 的更多信息

标签

Community Treasure Hunt

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

Start Hunting!

Translated by