pg_utils.numerics.matrices.invert_block_diag

pg_utils.numerics.matrices.invert_block_diag(matrix: ndarray, block_seg: List[int]) → ndarray[source]

Invert a block diagonal matrix

The matrix is assumed to be square matrix, and the block_seg gives the segmentation of the blocks on the diagonal.