diff --git a/mindspore/ops/operations/random_ops.py b/mindspore/ops/operations/random_ops.py
index c8d7c7576847a85fc78da6d6dfe99941b483c60d..18c2212b3d3aa6d90fa6e026891b916c26a723a3 100644
--- a/mindspore/ops/operations/random_ops.py
+++ b/mindspore/ops/operations/random_ops.py
@@ -45,7 +45,7 @@ class RandomChoiceWithMask(PrimitiveWithInfer):
 
     Examples:
         >>> rnd_choice_mask = P.RandomChoiceWithMask()
-        >>> input_x = Tensor(np.ones(shape=[240000, 4]), mindspore.bool_)
+        >>> input_x = Tensor(np.ones(shape=[240000, 4]).astype(np.bool))
         >>> output_y, output_mask = rnd_choice_mask(input_x)
     """