I am wondering is there an easy way to extract the diagonal with cuSparse? In Eigen, we can do
Eigen::VectorXd A_diag = A.diagonal();
Is there an equivalent for this in cuSparse? Or do I have to write my own kernel to do it?
I am wondering is there an easy way to extract the diagonal with cuSparse? In Eigen, we can do
Eigen::VectorXd A_diag = A.diagonal();
Is there an equivalent for this in cuSparse? Or do I have to write my own kernel to do it?