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

!29670 Fix doc for gumbel distribution 

Merge pull request !29670 from zichun_ye/doc_fix
This commit is contained in:
i-robot 2022-02-07 03:11:57 +00:00 committed by Gitee
parent 82e8436841 f2e8834d82
commit 451930ea5a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/api/api_python/nn_probability/mindspore.nn.probability.distribution.Gumbel.rst
View File

@ -13,8 +13,8 @@ mindspore.nn.probability.distribution.Gumbel
**参数：**
- **loc** (int, float, list, numpy.ndarray, Tensor) - Gumbel分布的位置。默认值：None。
- **scale** (int, float, list, numpy.ndarray, Tensor) - Gumbel分布的尺度。默认值：None。
- **loc** (int, float, list, numpy.ndarray, Tensor) - Gumbel分布的位置。
- **scale** (int, float, list, numpy.ndarray, Tensor) - Gumbel分布的尺度。
- **seed** (int) - 采样时使用的种子。如果为None，则使用全局种子。默认值：0。
- **dtype** (mindspore.dtype) - 分布类型。默认值：mindspore.float32。
- **name** (str) - 分布的名称。默认值：'Gumbel'。

4
mindspore/python/mindspore/nn/probability/distribution/gumbel.py
View File

@ -37,8 +37,8 @@ class Gumbel(TransformedDistribution):
where a and b are loc and scale parameter respectively.
Args:
loc (int, float, list, numpy.ndarray, Tensor): The location of Gumbel distribution. Default: None.
scale (int, float, list, numpy.ndarray, Tensor): The scale of Gumbel distribution. Default: None.
loc (int, float, list, numpy.ndarray, Tensor): The location of Gumbel distribution.
scale (int, float, list, numpy.ndarray, Tensor): The scale of Gumbel distribution.
seed (int): the seed used in sampling. The global seed is used if it is None. Default: 0.
dtype (mindspore.dtype): type of the distribution. Default: mstype.float32.
name (str): the name of the distribution. Default: 'Gumbel'.