ENH Updates docs and tests for regression modules that already support no-batch-dims (#61461)
Summary:
Towards https://github.com/pytorch/pytorch/issues/60585
This PR does not use `check_sum_reduction` because I wanted to test every reduction option.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61461
Reviewed By: suo
Differential Revision: D29883744
Pulled By: jbschlosser
fbshipit-source-id: cdad0effb41f0484938caad0d4c9d6d83e2aec07