固定代码

Question

我有一个带有正定矩阵的三维数组，我想获得一个与所有矩阵的Cholesky因子相同大小的数组。我正在使用Armadillo库和cube类型，其中有一个方便的功能each_slice，我试图使用它。但我没有让lambda表达式正常工作，所以希望有人可以帮助我并指出我的错误。

这是一个最小的例子：

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

// [[Rcpp::export]]
arma::cube chol_array(arma::cube Sigma) {
  arma::cube Sigma_chol = Sigma;
  Sigma_chol.each_slice([](arma::mat X) {return arma::chol(X);});
  return Sigma_chol;
}
// [[Rcpp::export]]
arma::cube chol_array2(arma::cube Sigma) {
  arma::cube Sigma_chol(size(Sigma));
  for (arma::uword i = 0; i < Sigma.n_slices; i++) {
    Sigma_chol.slice(i) = arma::chol(Sigma.slice(i));
  }
  return Sigma_chol;
}

/*** R
Sigma <- array(crossprod(matrix(rnorm(9), 3, 3)), dim = c(3, 3, 2))
chol_array(Sigma)
chol_array2(Sigma)
*/

函数chol_array2完成工作，但chol_array只返回原始矩阵。我错过了什么？

Answer 1

这里的问题是.each_slice()电话中没有引用。 Armadillo使用lambda表达式需要引用来更新对象，而不是一个return语句。特别是，我们有：

表格3：

将给定的lambda_function应用于每个切片; 该函数必须接受对与底层多维数据集具有相同元素类型的Mat对象的引用

所以，改变：

Sigma_chol.each_slice([](arma::mat X) {return arma::chol(X);});

为：

Sigma_chol.each_slice([](arma::mat& X) {X = arma::chol(X);});

固定代码

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

// Enable lambda expressions.... 
// [[Rcpp::plugins(cpp11)]]

// [[Rcpp::export]]
arma::cube chol_array(arma::cube Sigma) {
  arma::cube Sigma_chol = Sigma;

  // NOTE: the '&' and saving _back_ into the object are crucial
  Sigma_chol.each_slice( [](arma::mat& X) { X = arma::chol(X); } ); 

  return Sigma_chol;
}

测试代码

set.seed(1113)
Sigma = array(crossprod(matrix(rnorm(9), 3, 3)), dim = c(3, 3, 2))
all.equal(chol_array(Sigma), chol_array2(Sigma))
# [1] TRUE

RcppArmadillo：使用each_slice进行Lambda表达式

1 个答案:

固定代码

测试代码