Броненосец - норма каждого маленького блока в длинном векторе - PullRequest
Новая
       41

Броненосец - норма каждого маленького блока в длинном векторе

0 голосов
/ 04 марта 2019

Я использую Armadillo в C ++.

У меня длинный вектор с 10 элементами.Я хочу взять норму 2 каждого блока из 2 смежных значений.В итоге у меня будет 5 значений.

В RI можно преобразовать этот вектор в матрицу и использовать apply, но я не уверен, как это сделать в Armadillo.Цени любую помощь

1 Ответ

0 голосов
/ 23 апреля 2019

Вам просто нужно создать матрицу из вашего вектора, а затем перебрать столбцы.

#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
...