带有附加参数的RcppParallel worker。

Question

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

我试着紧贴着 并行 的例子，但我需要一个额外的（非迭代）参数，用于worker的 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 参数？

Answer 1

replacer 是一个函数 模板，而不是一个函数，这意味着除非使用特定的实例化，否则它不能作为函数对象使用，因为否则模板参数推导失败。

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

你可以使用一个通用的lambda表达式来代替。

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

这样的话，即使 replacer 是重载的或者是函数模板，这就可以了。

或者，删除 replacer 完全，直接使用lambda表达式。

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