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]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 2 New Failures

As of commit 931538c with merge base 98f8b37 ():

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

[scotts]

@5had3z, these changes are great, thanks for the PR!

[github-actions]

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