最近Cholmod
、いくつかの C++ コードでスパース コレスキー分解を実行するためにインストールしました。次に、分解を使用して逆行列を計算したかった(次の問題があります:
d^T . (A^-1 + B^-1)^-1 . d
ここd
で、ベクトル^T
は転置を示し、A
スパースB
です)
の実際の逆数を計算したかったのでB
、合計を含む解を線形に解くことができます。それを呼び出すコードは次のとおりです。
cholmod_common Common, *cm ;
cm = &Common ;
cholmod_start (cm) ;
cm->print=5;
Common.supernodal = CHOLMOD_SUPERNODAL ;
double *Tx, x;
int *Ti, *Tj, *Rdeg, *Cdeg,j;
cholmod_triplet *T ;
size_t nrow;
size_t ncol;
size_t nnz ;
nrow=Csize;
ncol=Csize;
nnz=numpulsars*numpulsars*numcoeff;
T = cholmod_allocate_triplet(nrow,ncol,nnz,0,CHOLMOD_REAL, cm) ;
Ti=(int*)T->i;
Tj=(int*)T->j;
Tx=(double*)T->x;
for(int i=0;i<numpulsars;i++){
for(int j=0;j<numpulsars;j++){
if(i==j){
pcorr=1.0;
}
else{
angle=pulsarCartesian[i][0]*pulsarCartesian[j][0] +pulsarCartesian[i][1]*pulsarCartesian[j][1]+pulsarCartesian[i][2]*pulsarCartesian[j][2];
pcorr=(1.5*(0.5*(1-angle))*log(0.5*(1-angle)) - 0.25*0.5*(1-angle) + 0.5);
}
for(int c1=0; c1<numcoeff; c1++){
val= pcorr*powercoeff[c1];
Ti[T->nnz]=i*numcoeff+c1;
Tj[T->nnz]=j*numcoeff+c1;
Tx[T->nnz]=val;
(T->nnz)++;
}
}
}
cholmod_sparse *PPFMSparse;
cholmod_factor *L ;
cholmod_dense *spinvK;
PPFMSparse=cholmod_triplet_to_sparse(T,T->nnz,cm);
// cholmod_print_sparse(PPFMSparse, "PPFMSparse", cm);
L = cholmod_analyze (PPFMSparse, cm) ;
cholmod_factorize (PPFMSparse, L, cm) ;
cholmod_sparse *PPFMinv;
PPFMinv=cholmod_spinv(L,cm);
// cholmod_print_sparse(PPFMinv, "PPFMinv", cm);
spinvK = cholmod_sparse_to_dense(PPFMinv, &Common) ;
cholmod_print_dense(spinvK, "spinvK", cm);
cholmod_free_sparse(&PPFMinv,cm);
cholmod_free_factor (&L, cm) ;
cholmod_free_sparse(&PPFMSparse,cm);
cholmod_free_triplet (&T, cm) ;
cholmod_free_dense (&spinvK, cm) ;
cholmod_finish(cm);
逆数を計算することを目的としたhttps://cholmod-extra.readthedocs.org/en/latest/functions.htmlこの関数を見つけましたが、逆数ではなく、逆数の2乗に関連する何かが得られます。誰かがこれを使用した経験があるかどうか、または C++ でスパース行列の逆数を計算するのと同等のことをしたかどうか疑問に思っていました。
乾杯リンドリー