Armadillo C ++：如何使用来自另一个矩阵的多个元素修改矩阵的多个数组元素，特别是在立方体结构中？

Question

在MATLAB中，如果我有两个3x3矩阵A和B并且想要将B的最后两列复制到A的前两个中，我将使用以下语法：

A(:,1:2) = B(:,2:3)

我试图在C ++中使用Armadillo完成相同的操作，特别是对于立方体结构。 在犰狳中，如果我有两个立方体A和B有九个切片（每个切片是一个3x3矩阵），我假设我将使用以下内容执行相同的列元素更新：

A(span(0,2),span(0,1),span(i)) = B(span(0,2),span(1,2),span(i))

其中'i'只是切片索引。 语法基于Armadillo语法指南。

代码编译没有错误并运行; 多维数据集切片只是没有更新。 我在这里使用正确的Armadillo语法，这是执行这些操作的最有效方法吗？

Answer 1

如果你想要一个简单的复制/粘贴（不是你的立方体矩阵的移位/旋转），你可以使用这种方法来解决这个问题：

#include <armadillo>

int main (int argc, char* argv[])
{
    size_t num_slices = 9;

    arma::icube A (3, 3, num_slices);
    arma::icube B (3, 3, num_slices);

    A.zeros();
    B.randn();

    A.print("Cube A :\n");
    B.print("Cube B :\n");

    for (int s = 0; s < num_slices; ++s)
        A.slice(s)(arma::span::all, arma::span(0, 1)) = 
        B.slice(s)(arma::span::all, arma::span(1, 2));

    A.print("Cube A :\n");
    B.print("Cube B :\n");

    return 0;
}