forked from mindspore-Ecosystem/mindspore
51 lines
1.7 KiB
ReStructuredText
51 lines
1.7 KiB
ReStructuredText
mindspore.nn.probability.distribution.Gumbel
|
||
================================================
|
||
|
||
.. py:class:: mindspore.nn.probability.distribution.Gumbel(loc, scale, seed=0, dtype=mindspore.float32, name='Gumbel')
|
||
|
||
耿贝尔分布（Gumbel distribution）。
|
||
|
||
**参数：**
|
||
|
||
- **loc** (float, list, numpy.ndarray, Tensor) - 耿贝尔分布的位置。
|
||
- **scale** (float, list, numpy.ndarray, Tensor) - 耿贝尔分布的尺度。
|
||
- **seed** (int) - 采样时使用的种子。如果为None，则使用全局种子。默认值：None。
|
||
- **dtype** (mindspore.dtype) - 分布类型。默认值：mindspore.float32。
|
||
- **name** (str) - 分布的名称。默认值：'Gumbel'。
|
||
|
||
**支持平台：**
|
||
|
||
``Ascend`` ``GPU``
|
||
|
||
.. note::
|
||
- `scale` 必须大于零。
|
||
- `dtype` 必须是浮点类型，因为耿贝尔分布是连续的。
|
||
- GPU后端不支持 `kl_loss` 和 `cross_entropy` 。
|
||
|
||
**样例：**
|
||
|
||
>>> import mindspore
|
||
>>> import mindspore.nn as nn
|
||
>>> import mindspore.nn.probability.distribution as msd
|
||
>>> from mindspore import Tensor
|
||
>>> class Prob(nn.Cell):
|
||
... def __init__(self):
|
||
... super(Prob, self).__init__()
|
||
... self.gum = msd.Gumbel(np.array([0.0]), np.array([[1.0], [2.0]]), dtype=mindspore.float32)
|
||
...
|
||
... def construct(self, x_):
|
||
... return self.gum.prob(x_)
|
||
>>> value = np.array([1.0, 2.0]).astype(np.float32)
|
||
>>> pdf = Prob()
|
||
>>> output = pdf(Tensor(value, dtype=mindspore.float32))
|
||
|
||
.. py:method:: loc
|
||
:property:
|
||
|
||
返回分布位置。
|
||
|
||
.. py:method:: scale
|
||
:property:
|
||
|
||
返回分布尺度。
|
||
|