blockJacobi.py 文件源码

def get_inversion(self):
        diag_blocks = self.blocks
        inverted_blocks = np.zeros(len(diag_blocks), dtype=object)
        for i in xrange(len(diag_blocks)):
            inverted_blocks[i] = np.linalg.inv(diag_blocks[i])

        inverted_diag = block_diag(*inverted_blocks)

        return inverted_diag