c++ - 如何在 Eigen 中对矩阵进行逐行裁剪？

Question

我需要在矩阵的每一行中剪辑(即钳制)元素。每列在双行矩阵中指定了最小值和最大值(第一行是最小值，第二行是最大值)。

我目前正在使用这种丑陋的方法来裁剪值。是否有更漂亮、性能更高的替代方案？

MatrixXf minmax {2, 3};
minmax <<
    0.4, 0.1, 1,   // minimums
    4.8, 4.0, 3.0; // maximums

MatrixXf mat = 5.0f * MatrixXf::Random(5,3);

// clamp each row according to minimums and maximums in minmax
for (int i = 0; i < mat.rows(); i++)
    mat.row(i) = mat.row(i).cwiseMin(minmax.row(1)).cwiseMax(minmax.row(0));

Answer 1

您可以使用广播:

mat = mat.cwiseMin(minmax.row(1).colwise().replicate(mat.rows()))
         .cwiseMax(minmax.row(0).colwise().replicate(mat.rows()));