【发布时间】:2015-05-12 06:05:40
【问题描述】:
访问犰狳矩阵的上三角或下三角元素的最有效(即平衡内存和速度)方法是什么?我知道我可以为元素提供一个整数向量,但是随着矩阵变得非常大,我想避免携带另一个大向量。或者有没有一种快速创建下/上三角指数的有效方法?
以 5x5 矩阵为例
// C++11 Initialization
arma::mat B = { 1, 2, 3, 4, 5,
6, 7, 8, 9, 10,
11, 12, 13, 14, 15,
16, 17, 18, 19, 20,
21, 22, 23, 24, 25 };
B.reshape(5,5);
// the matrix
//1 6 11 16 21
//2 7 12 17 22
//3 8 13 18 23
//4 9 14 19 24
//5 10 15 20 25
我想在结果向量所在的下三角形中提取元素:
2 3 4 5 8 9 10 14 15 20
我现在能想到的唯一解决方案是使用uvec 对象。例如:
arma::uvec idx {1,2,3,4,7,8,9,13,14,19);
arma::vec lower_elems = B.elem(idx);
最终对象不必是向量。我只需要能够访问元素进行各种比较。举个简单的例子,假设我想检查它们是否都等于 0。
【问题讨论】: