将 dnorm 与 RcppArmadillo 结合使用

Question

来自 R，我正在尝试运行 sourceCpp 此文件：

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

using namespace arma; 
using namespace Rcpp;

// [[Rcpp::export]]
vec dnormLog(vec x, vec means, vec sds) {
    int n = x.size();
    vec res(n);
    for(int i = 0; i < n; i++) {
        res[i] = log(dnorm(x[i], means[i], sds[i]));
    }
return res;
}

请参阅 this answer 了解我从哪里获得该功能。这会引发错误：

no matching function for call to 'dnorm4'

这正是我希望通过使用循环来防止的错误，因为引用的答案提到 dnorm 仅针对其第一个参数进行矢量化。我担心答案很明显，但我尝试在 dnorm 之前添加 R::，尝试使用 NumericVector 而不是 vec，而不在前面使用 log() .没有运气。但是，在 dnorm 之前添加 R:: 确实会产生一个单独的错误：

too few arguments to function call, expected 4, have 3; did you mean '::dnorm4'?

未通过将上面的dnorm替换为R::dnorm4来修复

Answer 1

这里有两个很好的教学时刻：

注意命名空间。如有疑问，请不要走向全球。
检查 headers 以获得实际定义。您错过了 scalar 版本 R::dnorm().

这是修复后的版本，其中包含您可能会感兴趣的第二个变体：

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

// [[Rcpp::export]]
arma::vec dnormLog(arma::vec x, arma::vec means, arma::vec sds) {
  int n = x.size();
  arma::vec res(n);
  for(int i = 0; i < n; i++) {
    res[i] = std::log(R::dnorm(x[i], means[i], sds[i], FALSE));
  }
  return res;
}

// [[Rcpp::export]]
arma::vec dnormLog2(arma::vec x, arma::vec means, arma::vec sds) {
  int n = x.size();
  arma::vec res(n);
  for(int i = 0; i < n; i++) {
    res[i] = R::dnorm(x[i], means[i], sds[i], TRUE);
  }
  return res;
}


/*** R
dnormLog( c(0.1,0.2,0.3), rep(0.0, 3), rep(1.0, 3))
dnormLog2(c(0.1,0.2,0.3), rep(0.0, 3), rep(1.0, 3))
*/

当我们获取这个时，return 得到相同的结果因为 R API 允许我们要求取对数。

R> sourceCpp("/tmp/dnorm.cpp")

R> dnormLog( c(0.1,0.2,0.3), rep(0.0, 3), rep(1.0, 3))
          [,1]
[1,] -0.923939
[2,] -0.938939
[3,] -0.963939

R> dnormLog2(c(0.1,0.2,0.3), rep(0.0, 3), rep(1.0, 3))
          [,1]
[1,] -0.923939
[2,] -0.938939
[3,] -0.963939
R>

将 dnorm 与 RcppArmadillo 结合使用

Using dnorm with RcppArmadillo

c++

r

rcpp