[QAT] Check the value of numel to avoid segfault (pytorch#81547)

Fixes pytorch#78123

### Original Result

Segmentation fault

### Result after fix

RuntimeError: numel is out of the bound of input tensor
Pull Request resolved: pytorch#81547
Approved by: https://github.com/kit1980

aten/src/ATen/native/quantized/QTensor.cpp

@@ -330,6 +330,10 @@ std::tuple<Tensor, Tensor> choose_qparams_optimized(
     const double ratio,
     int64_t bit_width) {
 
+  if (numel < 0 || numel > input_tensor.numel()) {
+    TORCH_CHECK(false, "numel is out of the bound of input tensor");
+  }
+
   TORCH_CHECK(numel <= input_tensor.numel(), "numel ", numel,
       " greater than input_tensor.numel() ", input_tensor.numel());
   const float* input_row = input_tensor.data_ptr<float>();