Fix the error in the example of `RandomChoiceWithMask` operation.

This commit is contained in:
seatea 2020-04-15 22:27:12 +08:00
parent 5519af5568
commit bbf5d4b499
1 changed files with 1 additions and 1 deletions

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