From bbf5d4b4998896ef2d24128d5fa41724ae02d194 Mon Sep 17 00:00:00 2001 From: seatea Date: Wed, 15 Apr 2020 22:27:12 +0800 Subject: [PATCH] Fix the error in the example of `RandomChoiceWithMask` operation. --- mindspore/ops/operations/random_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/random_ops.py b/mindspore/ops/operations/random_ops.py index c8d7c7576..18c2212b3 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) """ -- GitLab