OpInfo for nn.functional.softmax (#62077)
Summary:
This PR:
* Adds OpInfo for `softmax` and `nn.functional.softmax` (alias).
* Skip removal for `test_jit_alias_remapping` test of `log_softmax`.
Please see https://github.com/facebookresearch/functorch/issues/78 and https://github.com/pytorch/pytorch/issues/54261.
cc: mruberry zou3519 pmeier
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62077
Reviewed By: heitorschueroff
Differential Revision: D29990019
Pulled By: zou3519
fbshipit-source-id: 67476990b54a5dd824eed9d10236e118564f2501
