pytorch
0adf0705 - Use scatter_reduce to support masked reductions on sparse COO tensors (sum, prod, amin, amax)

Commit
2 years ago
Use scatter_reduce to support masked reductions on sparse COO tensors (sum, prod, amin, amax) Pull Request resolved: https://github.com/pytorch/pytorch/pull/75454 Approved by: https://github.com/cpuhrsch
Committer
Parents
Loading