pytorch
6fd41665 - Add _foreach_addc(div/mul)(_).Tensor (#88157)

Commit
2 years ago
Add _foreach_addc(div/mul)(_).Tensor (#88157) Support passing value scalars as a flat 1D Tensor. Currently we can only pass either an individual scalar or a ScalarList. Pull Request resolved: https://github.com/pytorch/pytorch/pull/88157 Approved by: https://github.com/ngimel, https://github.com/albanD
Author
Committer
Parents
Loading