Browse Source

add the output of example of multinomial operator.

tags/v1.1.0
wangshuide2020 5 years ago
parent
commit
618d31a480
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ops/composite/random_ops.py

+ 2
- 0
mindspore/ops/composite/random_ops.py View File

@@ -257,6 +257,8 @@ def multinomial(inputs, num_sample, replacement=True, seed=None):
Examples:
>>> input = Tensor([0, 9, 4, 0], mstype.float32)
>>> output = C.multinomial(input, 2, True)
>>> print(output)
[1 2]
"""
shape = P.Shape()
reshape = P.Reshape()


Loading…
Cancel
Save