c++ - 计算 Eigen::Matrix 的 sign() 的最佳方法

Question

由于 Eigen C++ 库不包含用于计算矩阵的 sign(x) 的内置方法，因此我正在寻找执行此操作的最佳方法。有关 sign() 的定义，请参阅 Matlab documentation ，虽然我真的不需要 0 元素的情况。我想出的方法如下。

Eigen::MatrixXf a = Eigen::MatrixXf(2,2);
a << -0.5, 1.0,
      0.3, -1.4;

// Temporary objects containing 1's and -1's
const Eigen::MatrixXi pos = Eigen::MatrixXi::Ones(a.rows(), a.cols());
const Eigen::MatrixXi neg = Eigen::MatrixXi::Ones(a.rows(), a.cols()) * -1;

// Actually filling of the matrix sign(a)
Eigen::MatrixXi a_sign = (a.array() >= 0).select(pos, neg);

std::cout << a << std::endl << std::endl;
std::cout << a_sign << std::end;

这是可行的，所以输出由

-0.5    1
 0.3 -1.4

-1  1
 1 -1

但是我想知道是否有更好的方法来做到这一点？创建两个临时矩阵看起来很麻烦，并且在处理非常大的矩阵时会变得相当慢。

Answer 1

怎么样

 X = X.array().sign();