c++ - Armadillo 和 C++ - 使用二维数组进行矩阵初始化

Question

我需要用 double 组初始化 Armadillo 矩阵。我在原始文档中找到了这个构造函数:

mat(*aux_mem, n_rows, n_cols, copy_aux_mem = true, strict = true) 

我在 SO 上找到了关于它的问题:

armadillo C++: matrix initialization from array

问题是，构造函数似乎只适用于一维数组的初始化，而不适用于二维数组。如果我尝试以这种方式使用它:

double **matrix = new double *[block_size];
for(int i = 0; i < block_size; i++) {
    matrix[i] = new double[block_size];
}

arma::mat arma_matrix( &matrix[0][0], matrix_size, matrix_size, true, true );

cout << "am: " << arma_matrix[1][0] << endl;

我得到错误:

fined_grain:103/init_function: In function ‘void place_user_fn_103(ca::Context&, ca::TokenList<double>&)’:
fined_grain:103/init_function:61:42: error: invalid types ‘double[int]’ for array subscript

那么，用二维数组初始化 Arma 矩阵的理想方法是什么？我更喜欢最快的解决方案，因为我需要使用大矩阵。

Answer 1

我快速浏览了 armadillo library documentation我看到了以下问题。

1. 您传递给 arma_matrix 的参数在句法上是正确的，但不正确。您需要使用:

   double *matrix = new double [block_size*block_size];
   arma::mat arma_matrix( matrix, block_size, block_size, true, true );
   

2. 访问元素的语法是:

   cout << "am: " << arma_matrix(1, 0) << endl;
   

   您还可以使用:

   int row = 1;
   int col = 0;
   cout << "am: " << arma_matrix[row*block_size+col] << endl;