Rcpp犰狳中列表组件的矩阵乘法

Matrix Multiplication of List Components in RcppArmadillo

本文关键字:组件 列表 Rcpp      更新时间:2023-10-16

我想在 RcppArmadillo 中创建一个函数,它将两个列表 a 和 b 作为输入,然后返回一个列表"out",其中包含 a 和 b 的相应组件的矩阵乘积作为组件:out(i( = a(i(*b(i((在 for 循环中(。也就是说,我事先知道列表 a 和 b 的所有组件都是矩阵。

这可以在 RcppArmadillo 中使用as<arma::mat>实现,如以下代码示例所示:

#include <RcppArmadillo.h>
using namespace Rcpp;
// [[Rcpp::depends("RcppArmadillo")]]
// [[Rcpp::export]]
List test_1(List a, List b){
List out(a.size());
  for (int i=0; i<a.size(); i++){
   out(i) = as<arma::mat>(a(i))*as<arma::mat>(b(i));
  }
  return out;
}

但是,因为这旨在用作MCMC算法的一部分,目的是使其更快,并且因为我也看了这篇文章:Rcpp 中列表中的索引元素

我正在徘徊,与上面的代码相比,使用ListOf<arma::mat>这样的用户指定的类是否会导致更快的计算。我试图使用Romain Francois在链接中给出的ListOf类模板对其进行基准测试。 但是,在尝试获取代码时,我无法使其出现错误">对'ListOf'的引用不明确"。此外,我试图在上面的链接中获取罗曼·弗朗索瓦的整个代码,但仍然得到同样的错误。因此,尽管我在创建自己的类模板方面没有经验,因为我在 Rcpp 和 C++ 中有点新手,但我无法在速度方面比较两种解决方案。

尽管如此,总结一下,我的主要问题是:

  1. 自定义类模板解决方案(换句话说,创建一个新类,您可以在其中先验地指定列表组件的类(是否是as<class>的唯一替代方法,以便在 RcppArmadillo 中实现列表组件的矩阵乘法?

  2. 如果它确实是唯一的选择,它会比使用as<class>更快吗?

您可能过于

复杂了。 从R,你可以得到SEXP,你以便宜的方式实例化 - 参见RcppArmadillo包中的基准集。

一般来说,剖析胜过猜想。 只需尝试不同的东西并测量它们...