使用 Rcpp 的行矩阵减去向量

Question

我主要是 R 用户，但我想学习 Rcpp 以提高我的编码（速度）。所以我开始玩 C++ 和 Rcpp，我想我会尝试编写以下简单的函数，它采用矩阵行（即第一行）并从中减去一个向量 (m[1,]-vec)。

我知道这听起来很傻很简单，但我无法让它发挥作用。

代码 arma::mat beta = Rcpp::as(beta_);
arma::vec y = Rcpp::as(y_);
arma::rowvec S= beta.row(0);

arma::vec d = S - y;
return Rcpp::wrap(d);
'
fun m vec fun(m,vec)
fun(m, vec) 中的错误：

在 R 中，这将被执行为，
m[1,]-vec
0 2 2

Answer 1

library(RcppArmadillo)
library(inline)
code <- '
arma::mat beta = Rcpp::as<arma::mat>(beta_);
arma::rowvec y = Rcpp::as<arma::rowvec>(y_);
arma::rowvec S= beta.row(0);
arma::rowvec d = S - y;
return Rcpp::wrap(d);
'
fun <- cxxfunction(signature(beta_ ="matrix",y_="numeric"),code, plugin="RcppArmadillo")
m <- matrix(1:9,3)
vec <- c(1,2,5)
fun(m,vec)