From f2e8834d82171f618fb81d5125cd1f15380f6a52 Mon Sep 17 00:00:00 2001 From: Zichun Ye Date: Sat, 29 Jan 2022 15:54:11 +0800 Subject: [PATCH] fix default value for gumbel distribution --- .../mindspore.nn.probability.distribution.Gumbel.rst | 4 ++-- .../python/mindspore/nn/probability/distribution/gumbel.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/api/api_python/nn_probability/mindspore.nn.probability.distribution.Gumbel.rst b/docs/api/api_python/nn_probability/mindspore.nn.probability.distribution.Gumbel.rst index 0e16944f67d..33f70bc498f 100644 --- a/docs/api/api_python/nn_probability/mindspore.nn.probability.distribution.Gumbel.rst +++ b/docs/api/api_python/nn_probability/mindspore.nn.probability.distribution.Gumbel.rst @@ -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'。 diff --git a/mindspore/python/mindspore/nn/probability/distribution/gumbel.py b/mindspore/python/mindspore/nn/probability/distribution/gumbel.py index ee3c9ea6f0e..67d6562b214 100644 --- a/mindspore/python/mindspore/nn/probability/distribution/gumbel.py +++ b/mindspore/python/mindspore/nn/probability/distribution/gumbel.py @@ -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'.