User Guide
API Reference
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.