rusty1s / pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations
MIT License
1.01k stars 147 forks source link

add element-wise mul for SparseTensor #238

Closed VeritasYin closed 1 year ago

VeritasYin commented 2 years ago

Hello,

As a followup of #236, I implemented the element-wise multiplication for two sparse matrices and added a test file for the mul operation.

Best, Haoteng

rusty1s commented 1 year ago

Merged via #323. Thank you!