diag

iskra.sparse.diag(values: Tensor | SparseTensor) → SparseTensor[SOURCE]

Constructs a sparse COO diagonal matrix from vector of diagonal elements.

Parameters:

values (Tensor[DType, [N]] | SparseTensor[DType, [N]]) – Vector with values that will be placed along the diagonal. Allowed to be either dense or sparse.

Returns:

​ (SparseTensor[DType, [N, N]]) – Sparse N-by-N identity matrix in COO format.