The usage of the three function exported by the package is very simple. First, let's create an example matrix whose diagonal is easy to estimate. n=20 Om = randn(n,n) Q,R = qr(Om) A_temp = Q*diagm(vec ...
This repository contains benchmarking results for different ways to extract diagonal entries from a sparse matrix in PyTorch. Python for-loop (CSR): Here, we iterate over the rows, select the ...