如何在犰狳或 RcppArmadillo 中将矩阵复制到立方体中

Question

我有一个维度为 (k1, k2) 的矩阵（比如 A）。我想将它复制 N 次并将其保存到一个名为 B 的 3D 数组或立方体中。结果，B 的维度将是 (k1, k2, N)。

在 R 中，我做了以下操作：

B <- replicate(N, A)

我想知道犰狳中是否有可以做到这一点的函数。或者，循环是唯一的方法吗？

谢谢！

Answer 1

您可以避免使用each_slice() 方法的循环，但仍必须事先初始化多维数据集：

arma::cube B(k1, k2, N);
B.each_slice() = A;

它的优点是简洁。