torch.where : Scalar Support (#40336)
Summary:
Reference: https://github.com/pytorch/pytorch/issues/38349 #9190
TODO
* [x] Add Tests
* [x] Update Docs
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40336
Reviewed By: albanD
Differential Revision: D22813834
Pulled By: mruberry
fbshipit-source-id: 67c1693c059a301b249213afee3c25cea9f64fec