如何使用C STL算法重写嵌套循环？

循环很简单,但是我似乎不能用我的头围绕着使用STL算法给出下面相同的嵌套循环.

const int a_size = 5; // input
const int c_size = 2; // output
const int b_size = a_size * c_size; // multipliers

std::vector<float> a(a_size);
std::vector<float> b(b_size);
std::vector<float> c(c_size);

// fill a and b with data

// this nested loop
for(int i = 0; i<c_size; i++) {
    c[i] = 0.0;
    for(int k = 0; k<a_size; k++) {
        c[i] += (a[k] * b[i*a_size+k]);
    }
    c[i] = sigmoid(c[i]);
}

我想这样做的原因是Boost.Compute库,它将使用类似STL的算法(std :: transform,std :: for_each等)对GPU进行计算.

解决方法

我来了 with：

auto i = 0;
generate(begin(c),end(c),[&i,&a,&b]
{
    return sigmoid(inner_product
    (
        begin(a),end(a),begin(b) + distance(begin(a),end(a)) * i++,0.f
    ));
});

但是看起来不是很好 – 在这种情况下,我更愿意编写自己的算法.

或使用矩阵形式.与Eigen图书馆将成为：

MatrixXd b;
VectorXd a,c;
// ...
c = (b*a).unaryExpr(sigmoid);