ENH Adds tests and docs for 2d & 3d modules that already support no batch (#61262)
Summary:
Toward https://github.com/pytorch/pytorch/issues/60585
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61262
Reviewed By: mrshenli
Differential Revision: D29660554
Pulled By: jbschlosser
fbshipit-source-id: d5e3dc7096fcf8621bce4a1063d521b84092e0ca