我需要 2 个 for 循环来填充 Eigen 中的矩阵，但我可以在 Matlab 中只用 1 个 for 循环填充它 - 我可以摆脱额外的 for 循环吗？

Question

我正在用以下代码填充特征矩阵：

int M = 3;
int N = 4;
MatrixXd A(M, N);

double res = sin(4);

for (int i = 0; i < M; i++) {
    for (int j = 0; j < N; j++) {
        A(i, j) = sin(i+j);
    }
}

在 Matlab 中，我只需要 1 个 for 循环就可以使用矢量化来做同样的事情：

M = 3;
N = 4;
N_Vec = 0:(N-1);
A = zeros(M,N);
for i=1:M
    A(i,:) = sin((i-1)+N_Vec);
end

是否可以在 C++/Eigen 中做类似的事情，这样我就可以摆脱其中一个 for 循环？ 如果有可能以某种方式摆脱这两个循环，那就更好了。 那可能吗？

Answer 1

使用NullaryExpr ，您可以使用 Eigen 中的零（手动）循环来执行此操作：

Eigen::MatrixXd A = Eigen::MatrixXd::NullaryExpr(M, N,
      [](Eigen::Index i, Eigen::Index j) {return std::sin(i+j);});

当使用优化编译时，它不一定比手动的双循环版本快（如果没有优化，它甚至可能更慢）。

您可以编写int或long而不是Eigen::Index ，如果这样更具可读性...