pytorch

f03e7170 - ENH Updates docs and tests for regression modules that already support no-batch-dims (#61461)

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