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