如何在matlab中迭代稀疏矩阵中的元素？

Question

我正在 matlab 中为 LDPC 信念传播解码器编写代码。 我遵循以下指定的算法： http://www.ece.umd.edu/~tta/resources/LDPC.pdf 第 6 页。

我想使用稀疏矩阵来实现这个目标。 在此期间，我想迭代只有少数几个的大型矩阵。 我的问题是如何只迭代矩阵中的那些？

我会更具体：假设我有以下 matlab 代码：

for row_index = 1 : row_num
    for coloum_index = 1 : col_num
        if parity_check_matrix(row_index, coloum_index)
            messages_llr_matrix(row_index, coloum_index) = ...
                code_word_aprior_prob(coloum_index);
        end
    end

此代码采用向量“code_word_aprior_prob”并将其值分配给矩阵“messages_llr_matrix”中非零的行。现在，我想使用稀疏矩阵来执行此操作。

parity_check_matrix 是一个稀疏矩阵。

Answer 1

假设你有一个稀疏矩阵S。您可以使用

迭代其非零元素

[ii,jj,ss] = find(S);

for k=length(ii)
   %// A nonzero element of S: ss(k) = S(ii(k),jj(k))
end

虽然我不确定这会有多快。

在您的具体情况下，您可能会这样做

[ii,jj] = find(parity_check_matrix);
for k = 1 : length(ii)
   messages_llr_matrix(ii(k), jj(k)) = ...
            code_word_aprior_prob(jj(k));
end

当然你可以将ii和jj重命名为row_index和column_index。

Answer 2

以下内容执行您的代码 sn-p 所做的事情，但并不完全迭代稀疏矩阵，就像问题的标题所暗示的那样。另外，我不确定我的其他答案中的方法是否比这更快或更慢。

请注意，我假设 code_word_aprior_prob 是一个行向量。然后做

messages_llr_matrix = bsxfun(@times, parity_check_matrix, code_word_aprior_prob)

可能就够了。

请测试这是否真的给出了正确的答案，如果您真的以速度为目标，请测试哪种方式更快。