如何在犰狳中加入矩阵

How to join matrices in Armadillo

我有几个进程,每个进程都计算自己的矩阵。最后,我希望他们将解出的矩阵发送到根进程,根进程将它们真正地连接成一个大矩阵,并以 ASCII 格式简单地将其输出到一个文本文件中。是否有可能以某种优雅的方式或我必须遍历每个矩阵并自己“加入”它?


示例:

matrix 0:
0 5 
4 5 

matrix 1:
1 2
3 4

matrix 2:
5 6
8 7

matrix 3:
0 0
2 1

Final "joined" matrix:
0 5 | 1 2
4 5 | 3 4
----+----
5 6 | 0 0
8 7 | 2 1

假设你的四个矩阵数据分别存储在data/a.txt,data/b.txt,data/c.txt,data/d.txt中,下面的代码会产生期望的结果。

arma::mat a,b,c,d ;

a.load("data/a.txt");
b.load("data/b.txt");
c.load("data/c.txt");
d.load("data/d.txt");

// join rows (=horizontally, i.e. matrices must have the same number of rows)
auto joined_ab  = std::move(arma::join_rows( a, b ));
auto joined_cd  = std::move(arma::join_rows( c, d ));

// join columns (=vertically, i.e. matrices must have the same number of cols)
auto joined_mat = std::move(arma::join_cols( joined_ab, joined_cd ));

std::cout << joined_mat << std::endl;

注意:对于原始问题,以下内容成立;

a = matrix 0

b = matrix 1

c = matrix 2

d = matrix 3

所以上面的构造产生:

a | b 
--+---
c | d