Fix Parametrizations tutorial

[svekars]

Fixes https://github.com/pytorch/tutorials/actions/runs/16273680500/job/45947334322?pr=3416#step:9:2088

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/3461

• 📄 Preview Python docs built from this PR

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

❌ 3 New Failures

As of commit 95aaf0b with merge base 9a44439 ():

NEW FAILURES - The following jobs have failed:

• Build tutorials / pytorch_tutorial_build_worker (2, 15, linux.g5.4xlarge.nvidia.gpu) (gh)
  RuntimeError: Expected input at *args[0].shape[0] to be equal to 1, but got 2. If you meant for this dimension to be dynamic, please re-export and specify dynamic_shapes (e.g. with Dim.DYNAMIC)
• Build tutorials / pytorch_tutorial_build_worker (6, 15, linux.g5.4xlarge.nvidia.gpu) (gh)
  Process completed with exit code 2.
• Build tutorials / pytorch_tutorial_build_worker (8, 15, linux.g5.4xlarge.nvidia.gpu) (gh)
  Process completed with exit code 2.

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