Allow all single-element tensors as value for ConstantOfShape

onnxruntime/core/providers/cpu/generator/constant_of_shape_base.h

@@ -78,8 +78,14 @@
     auto* t_proto_p = t_proto.get();
 #endif
     if (info.GetAttr<ONNX_NAMESPACE::TensorProto>("value", t_proto_p).IsOK()) {
-      ORT_ENFORCE(t_proto_p->dims_size() == 1, "Must have a single dimension");
-      ORT_ENFORCE(t_proto_p->dims()[0] == 1, "Must have a single dimension of 1");
+      ORT_ENFORCE(
+          t_proto_p->dims_size() == 0 ||
+              std::accumulate(
+                  t_proto_p->dims().begin(),
+                  t_proto_p->dims().end(),
+                  1,
+                  std::multiplies<size_t>()) == 1,
+          "The value attribute of ConstantOfShape must be a single-element tensor");
       SetValueFromTensorProto(*t_proto_p);
     } else {
       float f_value = 0.f;