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

code docs

This commit is contained in:
changzherui 2022-12-15 12:18:13 +08:00
parent 0a2607003a
commit 24aaf619c9
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docs/api/api_python/ops/mindspore.ops.func_choice_with_mask.rst
View File

@ -1,7 +1,7 @@
mindspore.ops.choice_with_mask
=====================================
.. py:function:: mindspore.ops.choice_with_mask(input_x, count=256, seed=0, seed2=0)
.. py:function:: mindspore.ops.choice_with_mask(input_x, count=256, seed=None)
对输入进行随机取样，返回取样索引和掩码。
@ -11,8 +11,7 @@ mindspore.ops.choice_with_mask
参数：
- **input_x** (Tensor[bool]) - 输入Tensor，bool类型。秩必须大于等于1且小于等于5。
- **count** (int) - 取样数量，必须大于0。默认值：256。
- **seed** (int) - 随机种子。默认值：0。
- **seed2** (int) - 随机种子2。默认值：0。
- **seed** (int) - 随机种子。默认值：None。
返回：
两个Tensor，第一个为索引，另一个为掩码。
@ -22,5 +21,5 @@ mindspore.ops.choice_with_mask
异常：
- **TypeError** - `count` 不是int类型。
- **TypeError** - `seed` 或 `seed2` 不是int类型。
- **TypeError** - `seed` 不是int类型。
- **TypeError** - `input_x` 不是Tensor。