mindspore/docs/api/api_python/ops/mindspore.ops.UniformInt.rst

36 lines
1.5 KiB
ReStructuredText
Raw Normal View History

2022-05-29 11:14:01 +08:00
mindspore.ops.UniformInt
========================
.. py:class:: mindspore.ops.UniformInt(seed=0, seed2=0)
2022-11-19 10:12:44 +08:00
根据均匀分布在区间 `[minval, maxval)` 中生成随机数。离散概率函数定义如下:
2022-05-29 11:14:01 +08:00
.. math::
\text{P}(i|a,b) = \frac{1}{b-a+1},
2022-06-16 17:12:08 +08:00
其中 :math:`a` 为分布区间的最小值 `minval` :math:`b` 为分布区间的最大值 `maxval`
2022-05-29 11:14:01 +08:00
.. note::
2023-02-28 11:32:27 +08:00
- `minval` 中的数值在广播后必须严格小于 `maxval`
- 如果 `seed``seed2` 都没有被赋非零值,则生成一个随机值当做随机种子
2022-05-29 11:14:01 +08:00
2022-07-22 11:20:32 +08:00
参数:
- **seed** (int) - 随机种子非负值。默认值0。
- **seed2** (int) - 随机种子2用来防止随机种子冲突非负值。默认值0。
输入:
2022-12-24 11:53:31 +08:00
- **shape** (Union[tuple, Tensor]) - 目标Tensor的shape。只允许常量值。
2022-07-22 11:20:32 +08:00
- **minval** (Tensor) - 分布参数, :math:`a`
决定可能生成的最小值数据类型为int32。需为标量。
- **maxval** (Tensor) - 分布参数, :math:`b`
决定生成随机数的上限数据类型为int32。需为标量。
输出:
Tensor。shape为输入 `shape` 数据类型支持int32。
2022-07-22 11:20:32 +08:00
异常:
- **TypeError** - `seed``seed2` 不是int类型。
2022-12-24 11:53:31 +08:00
- **TypeError** - `shape` 不是tuple或Tensor。
2022-07-22 11:20:32 +08:00
- **TypeError** - `minval``maxval` 不是Tensor。
- **ValueError** - `shape` 不是常量值。