permutation test

rok · Dec 13, 2023 · 6f3d756 · 6f3d756
1 parent 9c60b85
commit 6f3d756
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cpp/src/arrow/extension/fixed_shape_tensor.cc b/cpp/src/arrow/extension/fixed_shape_tensor.cc
@@ -392,6 +392,12 @@ Result<std::shared_ptr<DataType>> FixedShapeTensorType::Make(
     return Status::Invalid("dim_names size must match shape size. Expected: ",
                            shape.size(), " Got: ", dim_names.size());
   }
+  for (auto i : permutation) {
+    if (i < 0 || i >= static_cast<int64_t>(shape.size())) {
+      return Status::Invalid("permutation indices must be in [0, shape.size()). Got: ",
+                             i);
+    }
+  }
   const auto size = std::accumulate(shape.begin(), shape.end(), static_cast<int64_t>(1),
                                     std::multiplies<>());
   return std::make_shared<FixedShapeTensorType>(value_type, static_cast<int32_t>(size),