src.utils.sparse.sbmm_diag

src.utils.sparse.sbmm_diag(t1, t2)

Perform bmm and diagonal for sparse x dense -> dense. The diagonalized result is returned in vector tensor. This is a wrapper function and does not support gradient.