cat_diag

iskra.sparse.cat_diag(xs: Sequence[SparseTensor]) → SparseTensor[SOURCE]

Concatenates sparse COO matrice along a diagonal to make a block-diagonal matrix.

Parameters:

xs (Sequence[SparseTensor]) – Sequences of matrices to cocnatenate.

Returns:

​ (SparseTensor) – Block-diagonal matrix with xs embedded along the diagonal.