Squeeze on the dimension we have selected

[SaoirseARM]

Summary

• In decompose_select.py - a squeeze op is used to get final output shape. If we squeeze using the input rank and batch_size==1 this dimension will also be removed.
• Changed this to only squeeze on dimension selected.

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/8335

• 📄 Preview Python docs built from this PR

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

❌ 1 New Failure, 1 Pending, 2 Unrelated Failures

As of commit 5866ca7 with merge base 524ec78 ():

NEW FAILURE - The following job has failed:

• pull / unittest / macos / macos-job (gh)
  [ FAILED ] TensorPtrMakerTest.CreateRandTensorWithBFloatType

FLAKY - The following job failed but was likely due to flakiness present on trunk:

• trunk / test-models-macos (cmake, emformer_join, xnnpack-quantization-delegation, macos-m1-stable, 90) / macos-job (gh) (similar failure)
  ##[error]The operation was canceled.

BROKEN TRUNK - The following job failed but was present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• trunk / test-models-macos (cmake, softmax, portable, macos-m1-stable, 90) / macos-job (gh) (trunk failure)
  ##[error]The operation was canceled.

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

[SaoirseARM]

@pytorchbot label ciflow/trunk

[SaoirseARM]

@pytorchbot label 'partner: arm'

[SaoirseARM]

@pytorchbot label "topic: not user facing"

[zingo]

MacOS fails are unrelated