Rcpp中pow的矢量化指數

Question

Rcpp允許矢量化一些操作，這很好。但是對於pow，只有基數可以是一個向量，而不是指數。通常情況下，在編譯：

#include <Rcpp.h> 
using namespace Rcpp; 
// [[Rcpp::export]] 
NumericVector puissancedCpp(NumericVector base, double exp){ 
    return pow(base,exp); 
} 

作品，但不是：

#include <Rcpp.h> 
using namespace Rcpp; 
// [[Rcpp::export]] 
NumericVector puissancedCpp(NumericVector base, NumericVector exp){ 
    return pow(base,exp); 
} 

什麼是執行的推薦方式中的R這將是：在其他東西中間

c(0,1,2,3)^c(4,3,2,1) 

在C完成？

Answer 1

這裏有一個選項，假設你的編譯器支持C++ 11：

#include <Rcpp.h> 
// [[Rcpp::plugins(cpp11)]] 

// [[Rcpp::export]] 
std::vector<double> vpow(const std::vector<double>& base, const std::vector<double>& exp) { 
    std::vector<double> res(base.size()); 
    std::transform(base.begin(), base.end(), exp.begin(), res.begin(), 
        [&](double lhs, double rhs) -> double { 
        return std::pow(lhs, rhs); 
        }); 
    return res; 
} 


/*** R 

c(0,1,2,3)^c(4,3,2,1) 
#[1] 0 1 4 3 
vpow(0:3, c(4,3,2,1)) 
#[1] 0 1 4 3 

*/ 

---

如果你正在使用舊的編譯器工作時，你可以使用

double dpow(const double lhs, const double rhs) { 
    return std::pow(lhs, rhs); 
} 

// [[Rcpp::export]] 
std::vector<double> vpow98(const std::vector<double>& base, const std::vector<double>& exp) { 
    std::vector<double> res(base.size()); 
    std::transform(base.begin(), base.end(), exp.begin(), res.begin(), dpow); 
    return res; 
} 

Answer 2

一個實現這一如果您由於某種原因不能使用C++ 11，則可以選擇替代方案11

#include <Rcpp.h> 

using namespace Rcpp; 

// [[Rcpp::export]] 
NumericVector vecpow(const NumericVector base, const NumericVector exp) { 
    NumericVector out(base.size()); 
    std::transform(base.begin(), base.end(), 
       exp.begin(), out.begin(), ::pow); 
    return out; 
} 

/*** R 
vecpow(c(0:3), c(4:1)) 
***/ 

哪產生

R> Rcpp::sourceCpp("vecpow.cpp") 

R> vecpow(c(0:3), c(4:1)) 
[1] 0 1 4 3 
R>