🐛 [Bug] aten::unbind converter produces incorrect output dimensions

[mfeliz-cruise]

Bug Description

torch.unbind removes the dimension it splits on. The existing converter for aten::unbind preserves the dimension in its output. You can see the difference in the shapes below from the aten::unbind unit tests.

jit: [3, 4, 4]
trt: [1, 3, 4, 4]
jit: [3, 4, 4]
trt: [1, 3, 4, 4]

jit: [5]
trt: [5, 1]

To Reproduce

Steps to reproduce the behavior:

1. Remove the reshapes on the trt outputs from the aten::unbind unit test. You will see that the split dimension is preserved in the converter output and removed in the jit output.

Expected behavior

Environment

Build information about Torch-TensorRT can be found by turning on debug messages

• Torch-TensorRT Version (e.g. 1.0.0):
• PyTorch Version (e.g. 1.0):
• CPU Architecture:
• OS (e.g., Linux):
• How you installed PyTorch (conda, pip, libtorch, source):
• Build command you used (if compiling from source):
• Are you using local sources or building from archives:
• Python version:
• CUDA version:
• GPU models and configuration:
• Any other relevant information:

Additional context

[github-actions]

This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days

[peri044]

@mfeliz-cruise Can you let me know which test are you referring to ?
I ran this one https://github.com/pytorch/TensorRT/blob/main/tests/core/conversion/converters/test_select.cpp#L1176-L1201 and got

===== JIT RESULTS: [3, 4, 4]
===== TRT RESULTS: [3, 4, 4]

[mfeliz-cruise]

My fix for this was merged in September: #1373

I think this can be closed.