src.utils.sparse.sssbmm_diag

src.utils.sparse.sssbmm_diag(m1, m2)

Perform bmm and diagonal for sparse x sparse -> sparse. This is a wrapper function and does not support gradient.