pytorch
cc5b0165 - Implement `mul(dense, sparse), mul(sparse, dense)` for sparse COO tensors. (#81556)

Commit
3 years ago
Implement `mul(dense, sparse), mul(sparse, dense)` for sparse COO tensors. (#81556) As per title. Implemented with broadcasting and in-place support. Follow-up : Backward implementation. Pull Request resolved: https://github.com/pytorch/pytorch/pull/81556 Approved by: https://github.com/amjames, https://github.com/cpuhrsch
Author
Committer
Parents
Loading