能否用std::for_each逐列转换arma::mat矩阵？

要直接对arma::mat使用带并行策略的std::for_each逐列处理，核心是解决Armadillo原生迭代器按元素遍历的问题。以下是两种可行方案：

方案一：基于列索引的并行遍历（推荐，简单高效） #

利用列索引范围替代列迭代器，直接通过arma::mat::col()获取列的引用，配合std::for_each的并行执行策略实现并行处理。

C++20版本（利用std::views::iota生成索引） #

#include <armadillo>
#include <iostream>
#include <algorithm>
#include <execution>
#include <ranges>

int main()
{
    arma::mat v{4, 3, arma::randu};
    std::cout << "原矩阵:\n" << v << "\n\n";

    // 生成0到列数-1的索引范围，并行遍历每一列
    std::for_each(std::execution::par,
                  std::views::iota(0, v.n_cols).begin(),
                  std::views::iota(0, v.n_cols).end(),
                  [&v](int col_idx) {
                      // 对当前列执行转换操作，这里用ones替代实际计算
                      v.col(col_idx) = arma::ones(v.n_rows);
                  });

    std::cout << "转换后矩阵:\n" << v << "\n";
    return 0;
}

兼容C++17及更早版本（手动生成索引列表） #

如果无法使用C++20的范围特性，可以预先生成列索引的std::vector：

#include <armadillo>
#include <iostream>
#include <algorithm>
#include <execution>
#include <vector>

int main()
{
    arma::mat v{4, 3, arma::randu};
    std::cout << "原矩阵:\n" << v << "\n\n";

    std::vector<int> col_indices(v.n_cols);
    std::iota(col_indices.begin(), col_indices.end(), 0);

    std::for_each(std::execution::par,
                  col_indices.begin(),
                  col_indices.end(),
                  [&v](int col_idx) {
                      v.col(col_idx) = arma::ones(v.n_rows);
                  });

    std::cout << "转换后矩阵:\n" << v << "\n";
    return 0;
}

方案二：自定义列迭代器适配器（贴合原生迭代器体验） #

如果希望完全模拟std::vector<arma::vec>的迭代器行为，可以自定义一个迭代器，每次返回对应列的arma::colvec&（与arma::vec&兼容）。

迭代器适配器实现示例 #

#include <armadillo>
#include <iterator>

class ArmaMatColIterator {
public:
    using iterator_category = std::random_access_iterator_tag;
    using value_type = arma::colvec;
    using difference_type = ptrdiff_t;
    using pointer = arma::colvec*;
    using reference = arma::colvec&;

    ArmaMatColIterator(arma::mat& mat, ptrdiff_t col_idx) : mat_(mat), col_idx_(col_idx) {}

    reference operator*() { return mat_.col(col_idx_); }
    pointer operator->() { return &mat_.col(col_idx_); }

    ArmaMatColIterator& operator++() { ++col_idx_; return *this; }
    ArmaMatColIterator operator++(int) { auto temp = *this; ++col_idx_; return temp; }

    ArmaMatColIterator& operator--() { --col_idx_; return *this; }
    ArmaMatColIterator operator--(int) { auto temp = *this; --col_idx_; return temp; }

    ArmaMatColIterator operator+(difference_type n) const { return ArmaMatColIterator(mat_, col_idx_ + n); }
    ArmaMatColIterator operator-(difference_type n) const { return ArmaMatColIterator(mat_, col_idx_ - n); }

    difference_type operator-(const ArmaMatColIterator& other) const { return col_idx_ - other.col_idx_; }

    bool operator==(const ArmaMatColIterator& other) const { return col_idx_ == other.col_idx_ && &mat_ == &other.mat_; }
    bool operator!=(const ArmaMatColIterator& other) const { return !(*this == other); }
    bool operator<(const ArmaMatColIterator& other) const { return col_idx_ < other.col_idx_; }
    // 其余比较运算符可按需实现

private:
    arma::mat& mat_;
    ptrdiff_t col_idx_;
};

// 为arma::mat添加生成列迭代器的辅助函数
ArmaMatColIterator col_begin(arma::mat& mat) { return ArmaMatColIterator(mat, 0); }
ArmaMatColIterator col_end(arma::mat& mat) { return ArmaMatColIterator(mat, mat.n_cols); }

使用自定义迭代器的代码 #

#include <iostream>
#include <algorithm>
#include <execution>

int main()
{
    arma::mat v{4, 3, arma::randu};
    std::cout << "原矩阵:\n" << v << "\n\n";

    std::for_each(std::execution::par,
                  col_begin(v),
                  col_end(v),
                  [](arma::vec& col) {
                      col = arma::ones(col.n_elem);
                  });

    std::cout << "转换后矩阵:\n" << v << "\n";
    return 0;
}

注意事项 #

• 并行编译支持：使用std::execution::par需要编译器开启并行支持，例如GCC添加-fopenmp参数，MSVC添加/openmp参数。
• 列操作独立性：确保你的转换函数仅依赖当前列的数据，避免跨列的竞态条件，这是并行安全的前提。
• Armadillo列类型兼容：arma::mat::col()返回的arma::colvec是arma::vec的派生类，可直接作为arma::vec&使用。

内容的提问来源于stack exchange，提问作者Øystein S