Batched Box Ops

[5had3z]

Make many box ops batch-dim compatible by adding leading ellipses and fixing unsqueezing dimensions for broadcasting. Updated documentation with leading ellipses. Add test for batched calcultions by comparing the batched result and iterative result.

Also fixed some linting issue raised by flake8 with a missing space in some assert message.

[pytorch-bot[bot]]

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/9058

• :page_facing_up: Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

:x: 2 New Failures

As of commit 931538cd663c70782dec74db113e0d4fc111f5f9 with merge base 98f8b3757c0648724064ca95434b18281c43c5f6 ():

NEW FAILURES - The following jobs have failed:

• Build Aarch64 Linux Wheels / pytorch/vision / build-wheel-py3_9-cuda-aarch6412_9-aarch64 (gh) ImportError: /lib64/libm.so.6: version GLIBC_2.29' not found (required by /usr/local/cuda/lib64/libnvshmem_host.so.3)`
• Build Aarch64 Linux Wheels / pytorch/vision / upload / upload-wheel-py3_9-cuda-aarch6412_9-aarch64 (gh) Unable to download artifact(s): Artifact not found for name: pytorch_vision__3.9_cu129_aarch64

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[scotts]

@pytorchmergebot rebase

[github-actions[bot]]

Hey @scotts!

You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py