ENH Adds test and docs for modules that already support no batch dims (#62729)
Summary:
Towards https://github.com/pytorch/pytorch/issues/60585
Pull Request resolved: https://github.com/pytorch/pytorch/pull/62729
Reviewed By: H-Huang
Differential Revision: D30669546
Pulled By: jbschlosser
fbshipit-source-id: c771c98c1fd9d28fa984b72893585c738c736505
