forked from mindspore-Ecosystem/mindspore
Fix the error in the example of `RandomChoiceWithMask` operation.
This commit is contained in:
parent
5519af5568
commit
bbf5d4b499
|
@ -45,7 +45,7 @@ class RandomChoiceWithMask(PrimitiveWithInfer):
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
>>> rnd_choice_mask = P.RandomChoiceWithMask()
|
>>> 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)
|
>>> output_y, output_mask = rnd_choice_mask(input_x)
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue