犰狳矢量类上的 RcppArmadillo 样本

Question

我们一直在使用RcppArmadillo 中的sample 函数来随机采样NumericVector 对象。然而，我们注意到不可能在犰狳类型（vec 或uvec）上使用相同的函数。我们查看了 sample.h 文件中的函数定义，它看起来像是一个模板化函数，应该能够使用这些类型，但我们无法弄清楚如何让它与 Armadillo 类一起使用而不做来自Rcpp 库的NumericVector 或IntegerVector 类型的大量转换。

例如，我们将这个函数写在一个名为 try.cpp 的文件中。

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>
#include <RcppArmadilloExtensions/sample.h>

using namespace arma;
using namespace Rcpp;

// [[Rcpp::export]]
arma::uvec sample_index(const int &size){
    arma::uvec sequence = linspace<uvec>(0, size-1, size);
    arma::uvec out = sample(sequence, size, false);
    return out;
}

运行上面的代码会产生以下错误：

src/try.cpp|11 col 22 error| no matching function for call to 'sample' [cpp/gcc]      

~/Library/R/3.3/library/Rcpp/include/Rcpp/sugar/functions/sample.h|401 col 1 error| note: candidate function not viable: no known conversion from 'arma::uvec' (aka 'Col<unsigned int>') to 'int' for 1st argument [cpp/gcc]

~/Library/R/3.3/library/Rcpp/include/Rcpp/sugar/functions/sample.h|437 col 1 error| note: candidate template ignored: could not match 'Vector' against 'Col' [cpp/gcc]

对此的任何帮助将不胜感激:)

Answer 1

万一以后有人遇到这个问题，这个问题似乎与正在使用的命名空间中sample 函数的多个定义有关。专门键入定义所需功能的名称空间可以解决问题。特别是sample函数需要从Rcpp::RcppArmadillo调用。

以下代码可以按需要工作。

// [[Rcpp::depends(RcppArmadillo)]]
#include <RcppArmadillo.h>
#include <RcppArmadilloExtensions/sample.h>

// [[Rcpp::export]]
arma::uvec sample_index(const int &size){
    arma::uvec sequence = arma::linspace<arma::uvec>(0, size-1, size);
    arma::uvec out = Rcpp::RcppArmadillo::sample(sequence, size, false);
    return out;
}