repdiag

iskra.sparse.repdiag(x: SparseTensor, n_reps: int) → SparseTensor[SOURCE]

Repeats a sparse COO matrix along a diagonal to make a block-diagonal matrix.

Parameters:

• x (SparseTensor) – Matrix to repeat.

• n_reps (int) – Number of repetitions.

Returns:

​ (SparseTensor) – Block-diagonal matrix with x embedded along the diagonal n_reps times.