!365 Fix error in the example of `RandomChoiceWithMask`

Merge pull request !365 from seatea/randomchoicewithmask-example

!365 Fix error in the example of `RandomChoiceWithMask`
Merge pull request !365 from seatea/randomchoicewithmask-example
0a06d5e8 · mindspore-ci-bot · Gitee · 8f299228 · bbf5d4b4 · 0a06d5e8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mindspore/ops/operations/random_ops.py mindspore/ops/operations/random_ops.py +1 -1

未找到文件。
--- 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)
    """