From 66f1293a4783cc9412717b0fc68e583ce9d8356e Mon Sep 17 00:00:00 2001 From: baihuawei Date: Tue, 22 Sep 2020 08:57:04 +0800 Subject: [PATCH] modify note of type --- mindspore/nn/probability/distribution/categorical.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/probability/distribution/categorical.py b/mindspore/nn/probability/distribution/categorical.py index a8492ad52aa..6b0ee8eb002 100644 --- a/mindspore/nn/probability/distribution/categorical.py +++ b/mindspore/nn/probability/distribution/categorical.py @@ -29,7 +29,7 @@ class Categorical(Distribution): probs (Tensor, list, numpy.ndarray, Parameter): Event probabilities. logits (Tensor, list, numpy.ndarray, Parameter, float): Event log-odds. seed (int): The global seed is used in sampling. Global seed is used if it is None. Default: None. - dtype (mstype.int32): The type of the distribution. Default: mstype.int32. + dtype (mindspore.dtype): The type of the distribution. Default: mstype.int32. name (str): The name of the distribution. Default: Categorical. Note: