mindspore/docs/api/api_python/nn_probability/mindspore.nn.probability.di...

49 lines
1.5 KiB
ReStructuredText
Raw Normal View History

2021-12-02 14:56:03 +08:00
mindspore.nn.probability.distribution.Gumbel
================================================
2021-12-30 19:27:21 +08:00
.. py:class:: mindspore.nn.probability.distribution.Gumbel(loc, scale, seed=0, dtype=mstype.float32, name='Gumbel')
2021-12-02 14:56:03 +08:00
Gumbel分布Gumbel distribution
连续随机分布,取值范围为 :math:`(0, \inf)` ,概率密度函数为
.. math::
f(x, a, b) = 1 / b \exp(\exp(-(x - a) / b) - x).
其中 :math:`a, b` 为分别为Gumbel分布的位置参数和比例参数。
2021-12-02 14:56:03 +08:00
**参数:**
- **loc** (int, float, list, numpy.ndarray, Tensor) - Gumbel分布的位置。
- **scale** (int, float, list, numpy.ndarray, Tensor) - Gumbel分布的尺度。
2022-01-08 17:07:59 +08:00
- **seed** (int) - 采样时使用的种子。如果为None则使用全局种子。默认值0。
2021-12-02 14:56:03 +08:00
- **dtype** (mindspore.dtype) - 分布类型。默认值mindspore.float32。
- **name** (str) - 分布的名称。默认值:'Gumbel'。
.. note::
- `scale` 必须大于零。
- `dtype` 必须是浮点类型因为Gumbel分布是连续的。
2021-12-02 14:56:03 +08:00
- GPU后端不支持 `kl_loss``cross_entropy`
**异常:**
- **ValueError** - `scale` 中元素小于0。
- **TypeError** - `dtype` 不是float的子类。
2021-12-02 14:56:03 +08:00
.. py:method:: loc
:property:
返回分布位置。
**返回:**
Tensor, 分布的位置值。
2021-12-02 14:56:03 +08:00
.. py:method:: scale
:property:
返回分布比例。
**返回:**
Tensor, 分布的比例值。