如何在 c++ 函子中传递 lambda 函数



我在Rccp和Rccpparallel很陌生,我很难弄清楚我在哪里犯了错误。因此,我想创建一个函数,该函数在矩阵中并行执行幂元素。我正在遵循 rcppParallel 示例。

在一个内核上,代码可以编译并工作正常,但是当我尝试将 n 传递给下面的函子时,我遇到了以下错误。

capture of non-variable "Power::n"
"this" was not captured for this lambda function
invalid use of non-static data member "Power::n"

如果我在下面的函子中交换 n,它可以编译并正常工作。我错过了什么?R 代码:

library(Rcpp)
library(RcppParallel)
Sys.setenv("PKG_CXXFLAGS"="-std=c++11")
sourceCpp("lambdaPower.cpp")

λ功率.cpp

#include <Rcpp.h>
using namespace Rcpp;
#include <cmath>
#include <algorithm>
// [[Rcpp::export]]
NumericMatrix matrixPower(NumericMatrix orig, double n)
{
    // allocate the matrix we will return
    NumericMatrix mat(orig.nrow(), orig.ncol());
    // transform it
    std::transform(orig.begin(), orig.end(), mat.begin(), [n](double x) { return pow(x, n); });
    // return the new matrix
    return mat;
}
// [[Rcpp::depends(RcppParallel)]]
#include <RcppParallel.h>
using namespace RcppParallel;
struct Power : public Worker
{
    // source matrix
    const RMatrix<double> input;
    // destination matrix
    RMatrix<double> output;
    //power
    double n;

    // initialize with source and destination
    Power(const NumericMatrix input, NumericMatrix output, double n)
        : input(input), output(output), n(n){}
    // take the n power of the range of elements requested
    void operator()(std::size_t begin, std::size_t end)
    {
        std::transform(input.begin() + begin,
                       input.begin() + end,
                       output.begin() + begin,
                       [n](double x) { return pow(x,n); }); // why n doesn work?
                    // If i swap n with fixed number it compiles and works.
                    // [](double x) { return pow(x,2); }); compiles and works
    }
};
// [[Rcpp::export]]
NumericMatrix parallelMatrixPower(NumericMatrix x, double n)
{
    // allocate the output matrix
    NumericMatrix output(x.nrow(), x.ncol());
    // power functor (pass input and output matrixes)
    Power power(x, output, n);
    // call parallelFor to do the work
    parallelFor(0, x.length(), power);
    // return the output matrix
    return output;
}

多谢。

如果将n复制到定义 lambda 的作用域中,则代码将编译:

....
  void operator()(std::size_t begin, std::size_t end)
  {
    auto _n = n;
    std::transform(input.begin() + begin,
                   input.begin() + end,
                   output.begin() + begin,
                   [_n](double x) { return pow(x,_n); });
  }
....

我不太擅长解释这一点,但您可以在斯科特·迈耶斯(Scott Meyers(的"有效现代C++"的"项目31:避免默认捕获模式"中阅读详细信息。

顺便说一句,我将在 C++ 代码中使用// [[Rcpp::plugins(cpp11)]],而不是在 R 代码中Sys.setenv("PKG_CXXFLAGS"="-std=c++11")

相关内容

  • 没有找到相关文章

最新更新