gammy.utils.extract_diag_blocks

extract_diag_blocks

extract_diag_blocks(x: numpy.ndarray, y: list) → List[numpy.ndarray][source]

Extract diagonal blocks from a matrix according to a reference