!7006 fix bugs of op DivNoNan, Div, MSSSIM and DropoutDoMask

Merge pull request !7006 from lihongkang/v2_master

mindspore/nn/layer/image.py

@@ -271,7 +271,7 @@ class MSSSIM(Cell):
         - **img2** (Tensor) - The second image batch with format 'NCHW'. It must be the same shape and dtype as img1.
 
     Outputs:
-        Tensor, has the same dtype as img1. It is a 1-D tensor with shape N, where N is the batch num of img1.
+        Tensor, the value is in range [0, 1]. It is a 1-D tensor with shape N, where N is the batch num of img1.
 
     Examples:
         >>> net = nn.MSSSIM(power_factors=(0.033, 0.033, 0.033))

mindspore/ops/operations/math_ops.py

@@ -1765,9 +1765,6 @@ class Div(_MathBinaryOp):
         Tensor, the shape is the same as the one after broadcasting,
         and the data type is the one with higher precision or higher digits among the two inputs.
 
-    Raises:
-        ValueError: When `input_x` and `input_y` do not have the same dtype.
-
     Examples:
         >>> input_x = Tensor(np.array([-4.0, 5.0, 6.0]), mindspore.float32)
         >>> input_y = Tensor(np.array([3.0, 2.0, 3.0]), mindspore.float32)
@@ -1806,9 +1803,6 @@ class DivNoNan(_MathBinaryOp):
         Tensor, the shape is the same as the one after broadcasting,
         and the data type is the one with higher precision or higher digits among the two inputs.
 
-    Raises:
-        ValueError: When `input_x` and `input_y` do not have the same dtype.
-
     Examples:
         >>> input_x = Tensor(np.array([-1.0, 0., 1.0, 5.0, 6.0]), mindspore.float32)
         >>> input_y = Tensor(np.array([0., 0., 0., 2.0, 3.0]), mindspore.float32)

mindspore/ops/operations/nn_ops.py

@@ -2418,8 +2418,8 @@ class DropoutDoMask(PrimitiveWithInfer):
         - **mask** (Tensor) - The mask to be applied on `input_x`, which is the output of `DropoutGenMask`. And the
           shape of `input_x` must be the same as the value of `DropoutGenMask`'s input `shape`. If input wrong `mask`,
           the output of `DropoutDoMask` are unpredictable.
-        - **keep_prob** (Tensor) - The keep rate, greater than 0 and less equal than 1, e.g. keep_prob = 0.9,
+        - **keep_prob** (Union[Tensor, float]) - The keep rate, greater than 0 and less equal than 1, e.g. keep_prob =
-          means dropping out 10% of input units. The value of `keep_prob` is the same as the input `keep_prob` of
+          0.9, means dropping out 10% of input units. The value of `keep_prob` is the same as the input `keep_prob` of
           `DropoutGenMask`.
 
     Outputs:
@@ -2463,7 +2463,11 @@ class DropoutDoMask(PrimitiveWithInfer):
 
         keep_prob_v = keep_prob['value']
         if keep_prob_v is not None:
-            validator.check_number_range('keep_prob', keep_prob_v.asnumpy(), 0, 1, Rel.INC_BOTH, self.name)
+            if isinstance(keep_prob['dtype'], type(mstype.tensor)):
+                validator.check_number_range('keep_prob', keep_prob_v.asnumpy(), 0, 1, Rel.INC_BOTH, self.name)
+            else:
+                validator.check_value_type("keep_prob", keep_prob_v, [float], self.name)
+                validator.check_number_range('keep_prob', keep_prob_v, 0, 1, Rel.INC_BOTH, self.name)
 
         out = {'shape': input_x_shape,
                'dtype': input_x['dtype'],