c++ - 犰狳 - 长向量中每个小块的范数

Question

我在 C++ 中使用犰狳。

我有一个包含 10 个元素的长向量。我想取每块 2 个相邻值的范数 2。最后，我将有 5 个值。

在 RI 中可以将该向量转换为矩阵并使用 apply 但我不确定如何在 Armadillo 中执行此操作。感谢任何帮助

Answer 1

你只需要从你的向量创建一个矩阵，然后遍历列。

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


// [[Rcpp::export]]
arma::vec foo_Cpp(arma::vec x) {
// Note that the dimension of x must be divisible by two.
  arma::mat X = arma::mat(x.memptr(), 2, x.n_elem/2);
  arma::uword n = X.n_cols;
  arma::vec norms = arma::vec(n);
  for (arma::uword i = 0; i < n; i++) {
    norms(i) = arma::norm(X.col(i), 2);
  }
  return norms;
}



/*** R
foo_R <- function(x) {
  X <- matrix(x, 2, length(x)/2)
  apply(X, 2, norm, type = "2")
}
x <- rnorm(1000)
all.equal(foo_R(x), c(foo_Cpp(x)))
microbenchmark::microbenchmark(foo_R(x), foo_Cpp(x))
*/

> all.equal(foo_R(x), c(foo_Cpp(x)))
[1] TRUE

> microbenchmark::microbenchmark(foo_R(x), foo_Cpp(x))
Unit: microseconds
       expr       min       lq        mean     median        uq       max neval
   foo_R(x) 17907.290 19640.24 21548.06789 20386.5815 21212.609 50780.584   100
 foo_Cpp(x)     5.133     6.34    26.48266    19.4705    21.734  1191.124   100