ENH Updates docs and tests for activation modules for no-batch dims (#61300)
Summary:
Towards https://github.com/pytorch/pytorch/issues/60585
This PR updates docs and tests for activation modules that already support no-batch dims.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61300
Reviewed By: heitorschueroff
Differential Revision: D29660543
Pulled By: jbschlosser
fbshipit-source-id: 5edad45f7e9995aca6c3403469668e6e1cbb94b6