p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!365 Fix error in the example of `RandomChoiceWithMask` 

Merge pull request !365 from seatea/randomchoicewithmask-example
This commit is contained in:
mindspore-ci-bot 2020-04-16 08:46:13 +08:00 committed by Gitee
parent 8f29922860 bbf5d4b499
commit 0a06d5e881
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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