Browse Source

!365 Fix error in the example of `RandomChoiceWithMask`

Merge pull request !365 from seatea/randomchoicewithmask-example
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 6 years ago
parent
commit
0a06d5e881
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ops/operations/random_ops.py

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

@@ -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)
"""



Loading…
Cancel
Save