带有附加参数的 RcppParallel worker

RcppParallel worker with additional arguments

这是我第一次尝试使用 RcppParallel 包,我必须使用 C++17 (Ubuntu)

我尽量接近开发者网站的 ParallelFor 示例,但我需要为工作人员提供额外的(非迭代)参数 threshold

这是我当前的代码

struct ReplaceWorker : public Worker
{
  // source matrix
  const RMatrix<double> input;

  // destination matrix
  RMatrix<double> output;

  // threshold
  double th;

  // initialize with source and destination
  ReplaceWorker(const NumericMatrix input, NumericMatrix output, double threshold) 
    : input(input), output(output), th(threshold) {}

  // replace function
  template<typename T>
  double replacer(const T &x){
    if(x < th){
      return(0);
    } else {
      return(1);
    }
  }

  // take the square root 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, 
                   replacer);
  }
};

然而我总是以相同的编译错误结束:

usr/include/c++/7/bits/stl_algo.h:4295:5: note: candidate: template<class _IIter, class _OIter, class _UnaryOperation> _OIter std::transform(_IIter, _IIter, _OIter, _UnaryOperation)
        transform(_InputIterator __first, _InputIterator __last,
        ^~~~~~~~~
   /usr/include/c++/7/bits/stl_algo.h:4295:5: note:   template argument deduction/substitution failed:
   network_edge_strength.cpp:173:28: note:   couldn't deduce template parameter ‘_UnaryOperation’
                       replacer);
                               ^
/usr/include/c++/7/bits/stl_algo.h:4332:5: note: candidate: template<class _IIter1, class _IIter2, class _OIter, class _BinaryOperation> _OIter std::transform(_IIter1, _IIter1, _IIter2, _OIter, _BinaryOperation)
        transform(_InputIterator1 __first1, _InputIterator1 __last1,
        ^~~~~~~~~
   /usr/include/c++/7/bits/stl_algo.h:4332:5: note:   template argument deduction/substitution failed:
   network_edge_strength.cpp:173:28: note:   candidate expects 5 arguments, 4 provided
                       replacer);
                               ^

任何建议,如何解决这个问题或替代方案,如何使用所需的 threshold 参数使其 运行?

replacer 是函数 template,不是函数,这意味着它不能用作函数对象,除非使用特定的实例化,否则模板参数推演失败

此外,作为成员函数,它需要一个隐式对象参数才能调用。

您可以改用通用的 lambda 表达式:

std::transform(/* [...] */, [this] (const auto& x) { return replacer(x); });

这样,即使 replacer 被重载或者是一个函数模板,它也可以工作。

或者,完全删除 replacer,并直接使用 lambda 表达式:

std::transform(/* [...] */, [this] (const auto& x) { return x < th ? 0 : 1; });