ENH Updates docs and tests for classification modules that already support no batch dims (#61874)
Summary:
Towards https://github.com/pytorch/pytorch/issues/60585
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61874
Reviewed By: heitorschueroff
Differential Revision: D29979977
Pulled By: jbschlosser
fbshipit-source-id: 82c19151aa7220e564337b05d7677d52981e0aa2