fix bug stack when dim -1

Signed-off-by: hongwei03 <[email protected]>

Author: p517332051

core/conversion/converters/impl/stack.cpp

@@ -19,9 +19,16 @@ auto stack_registrations TORCHTRT_UNUSED = RegisterNodeConversionPatterns().patt
      [](ConversionCtx* ctx, const torch::jit::Node* n, args& args) -> bool {
        auto in = args[0].IValue()->toListRef();
        auto dim = args[1].unwrapToInt();
+       if (-1 == dim) {
+         auto first_in = in[0];
+         if (first_in.isTensor()) {
+           dim = first_in.toTensor().ndimension();
+         } else {
+           dim = first_in.toCustomClass<TensorContainer>()->tensor()->getDimensions().nbDims;
+         }
+       }
 
        std::vector<nvinfer1::ITensor*> tensors;
-
        for (auto t : in) {
          nvinfer1::ITensor* itensor;