From 0ac47f2f71bb0780c29e641ecebeb354e0f64211 Mon Sep 17 00:00:00 2001
From: jiangjinsheng <jiangjinsheng@huawei.com>
Date: Fri, 29 May 2020 11:04:17 +0800
Subject: [PATCH] fixed ScalarCast

---
 mindspore/ops/operations/inner_ops.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mindspore/ops/operations/inner_ops.py b/mindspore/ops/operations/inner_ops.py
index ae7af9cd0..ad2ae3e95 100644
--- a/mindspore/ops/operations/inner_ops.py
+++ b/mindspore/ops/operations/inner_ops.py
@@ -15,6 +15,9 @@
 
 """inner_ops"""
 
+import numbers
+from ..._checkparam import Validator as validator
+from ..._checkparam import Rel
 from ...common.dtype import tensor, dtype_to_pytype
 from ..primitive import prim_attr_register, PrimitiveWithInfer
 
@@ -40,8 +43,10 @@ class ScalarCast(PrimitiveWithInfer):
         pass
 
     def __infer__(self, x, t):
+        validator.check_integer('x shape', len(x['shape']), 0, Rel.EQ, self.name)
         value, to = x['value'], t['value']
         if value is not None:
+            validator.check_value_type("value", value, [numbers.Number, bool], self.name)
             if isinstance(to, type(tensor)):
                 to = to.element_type()
             np_type = dtype_to_pytype(to)
-- 
GitLab