[complex] torch.sqrt: fix edge values (#47424)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/47358
Replace the optimized path with a slower but correct `map(std::sqrt)`
Benchmark posted below in comments.
cc: dylanbespalko (original author of fast-path)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47424
Reviewed By: walterddr
Differential Revision: D24855914
Pulled By: mruberry
fbshipit-source-id: c21a38f365d996645db70be96ff1216776bedd3a