Làm thế nào để viết lại vòng lặp lồng nhau bằng thuật toán C++ STL?

Question

Vòng lặp đủ đơn giản, nhưng tôi dường như không thể quấn quanh đầu tôi bằng cách sử dụng các thuật toán STL để cung cấp cùng một vòng lặp lồng nhau bên dưới.

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]); 
} 

Lý do tại sao tôi muốn làm điều này, là cho các thư viện Boost.Compute, mà sẽ làm các tính toán trên GPU sử dụng các thuật toán STL-like (std :: biến đổi, std :: for_each, vv .).

Answer 1

tôi đã đưa ra 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 
    )); 
}); 

Nhưng nó không giống khá tốt - có lẽ trong trường hợp như vậy tôi muốn viết thuật toán của riêng tôi.

Hoặc sử dụng biểu mẫu ma trận. Với thư viện Eigen, thư viện này sẽ trở thành:

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

Answer 2

thực tế vòng lặp lồng nhau là thuật toán std :: inner_product.

auto first = std::begin(b); 
auto increment = std::distance(std::begin(a), std::end(a)); 
//,, 

c[i] = std::inner_product(std::begin(a), std::end(a), first, 0); 
std::advance(first, increment); 

Thay vì vòng ngoài bạn có thể sử dụng thuật toán std :: generate.