ENH Adds dtype to nn.functional.one_hot (#58090)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/33046
Related to https://github.com/pytorch/pytorch/issues/53785
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58090
Reviewed By: zou3519
Differential Revision: D28640893
Pulled By: jbschlosser
fbshipit-source-id: 3686579517ccc75beaa74f0f6d167f5e40a83fd2
