dspdmm

hypercoil.functional.sparse.dspdmm(input: BCOO, diag: Tensor) → BCOO

Left and right matrix multiplication of a top-k format sparse matrix with a diagonal matrix. Returns a sparse matrix in the top-k format. For a sparse matrix X and a diagonal matrix D, the result is

\[\mathbf{D} \mathbf{X} \mathbf{D}\]