[special] special alias for softmax (#62251)
Summary:
Reference: https://github.com/pytorch/pytorch/issues/50345
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62251
Reviewed By: H-Huang
Differential Revision: D31141834
Pulled By: mruberry
fbshipit-source-id: aecaf62af248e9034ef589159ce0fb325c729493