微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何在 C++ 中使用 Eigen lib 创建块对角稀疏矩阵?

如何解决如何在 C++ 中使用 Eigen lib 创建块对角稀疏矩阵?

我想使用 Eigen lib 创建一个类似于下面的成本矩阵,其中 Q、QN 和 R 是一些方阵,N 是一个数字,即 20。

from scipy import sparse

P = sparse.block_diag([sparse.kron(sparse.eye(self.N),self.Q),self.QN,sparse.kron(sparse.eye(self.N),self.R)],format='csc')

我在 Eigen 库中找不到任何合适的 API。下面是我用来构造P矩阵的方法

Eigen::SparseMatrix<double> cost_;
cost_.resize(N,N);
cost_.setIdentity();


Eigen::SparseMatrix<double> stateCost =  Eigen::kroneckerProduct(cost_,model.Q).eval();
Eigen::SparseMatrix<double> controlCost =  Eigen::kroneckerProduct(cost_,model.R).eval();
Eigen::SparseMatrix<double> P;
P.resize((N+1)*numState+numInput*N,(N+1)*numState+numInput*N);
P.setZero();

for(int i=0; i<(N)*numState; i+=numState){
    for(int j=0; j<numState; j++){
        for(int k=0; k<numState; k++){
            P.coeffRef(i+j,i+k) = stateCost.coeffRef(i+j,i+k);
        }
    }
}
for(int i=(N)*numState; i<(N+1)*numState; i+=numState){
    for(int j=0; j<numState; j++){
        for(int k=0; k<numState; k++){
            P.coeffRef(i+j,i+k) =  model.QN.coeffRef(j,k);
        }
    }
}
int index = 0;
for(int i=(N+1)*numState; i<(N+1)*numState+numInput*N; i+=numInput){
    for(int j=0; j<numInput; j++){
        for(int k=0; k<numInput; k++){
            P.coeffRef(i+j,i+k) = controlCost.coeffRef(index+j,index+k);
        }
    }
index += numInput;
}

如果有其他有效的方法,请告诉我?我找到了这个 post,似乎真的很旧了,

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。