c++ - 如何使用 Eigen 在函数中访问/写入矩阵？

Question

我正在尝试使用 Eigen 编写一个函数来反转对角矩阵，这与平时有点不同。当对角线的元素为零(或比较接近于零)时，应将对角线元素的值设置为零，否则该值应为1/(对应元素)。我试图编写一个函数来接收我想要反转的对角矩阵(它实际上是一个 nx1 矩阵，因此得名)和另一个指针，我希望将结果放入其中:

template <typename m1, typename m2>
void invertSingularValues(Eigen::EigenBase<m1>& sing_val_vector,Eigen::EigenBase<m2>& res)
{
    for (int i=0; i<sing_val_vector.rows();i++)
        res(i,i)=(sing_val_vector[i]<0.0000001?0:1/sing_val_vector[i]);
};

似乎我无法使用 (i,j) 或 [i] 访问矩阵的元素，因为我收到此错误:

no match for ‘operator[]’ (operand types are ‘Eigen::EigenBase >’ and ‘int’) res(i,i)=(sing_val_vector[i]<0.0000001?0:1/sing_val_vector[i]);

no match for ‘operator[]’ (operand types are ‘Eigen::EigenBase >’ and ‘int’) res(i,i)=(sing_val_vector[i]<0.0000001?0:1/sing_val_vector[i]);

no match for call to ‘(Eigen::EigenBase >) (int&, int&)’ res(i,i)=(sing_val_vector[i]<0.0000001?0:1/sing_val_vector[i]);

当我这样调用函数时:

invertSingularValues(S.data,S_inv);

S.data 和 S_inv 是 Eigen 矩阵。 我能做什么？

Answer 1

正如编译器所说，当 res 的类型为 EigenBase 时，您不能执行 res(i,i)，因为 EigenBase 没有定义此功能。您需要使用更具体的类，例如 MatrixBase 或 DenseBase。

相关文档:

• Writing functions taking Eigen types as parameters
• The class hierarchy

忽略任何代码优化，这样的事情会起作用:

template <typename m1, typename m2>
void invertSingularValues(Eigen::MatrixBase<m1>& sing_val_vector,Eigen::MatrixBase<m2>&    res)
{ 
     for (int i(0); i < sing_val_vector.rows(); i++) 
         for (int j(0); j < sing_val_vector.cols(); j++) 
             res(i,j) = sing_val_vector(i,j) < 1e-07 ? 0.0 : 1.0/sing_val_vector(i,j);
};