Allow all single-element tensors as value for ConstantOfShape

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

@@ -78,8 +78,9 @@
     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");
+      for (auto dim : t_proto_p->dims()) {
+        ORT_ENFORCE(dim == 1, "The value attribute of ConstantOfShape must be a single-element tensor");
+      }
       SetValueFromTensorProto(*t_proto_p);
     } else {
       float f_value = 0.f;