Browse Source

!10351 add the output of example of multinomial operator.

From: @wangshuide2020
Reviewed-by: @liangchenghui,@wuxuejian
Signed-off-by: @liangchenghui
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
ac2e422af8
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