Make new_ones an operator (#58405)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/58394
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58405
Reviewed By: HDCharles
Differential Revision: D28480075
Pulled By: Chillee
fbshipit-source-id: bd29399867e2a002a2f395554621761d3c701f68