pyacs.lib.glinalg.extract_block_diag

Extract block diagonal from a matrix.

pyacs.lib.glinalg.extract_block_diag.extract_block_diag(a, n, k=0)[source]

Extract block diagonal from a 2D array.

Parameters:

a (array_like) – 2D array.
n (int) – Block size.
k (int, optional) – Diagonal offset. Default is 0.

Returns:

Block diagonal elements as 3D array of shape (n_blocks, n, n).

Return type:

numpy.ndarray