Browse Source

!6691 modify distribution type note

Merge pull request !6691 from baihuawei/distributiontype
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
990524c0e2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/probability/distribution/categorical.py

+ 1
- 1
mindspore/nn/probability/distribution/categorical.py View File

@@ -29,7 +29,7 @@ class Categorical(Distribution):
probs (Tensor, list, numpy.ndarray, Parameter): Event probabilities. probs (Tensor, list, numpy.ndarray, Parameter): Event probabilities.
logits (Tensor, list, numpy.ndarray, Parameter, float): Event log-odds. 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. 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. name (str): The name of the distribution. Default: Categorical.
Note: Note:


Loading…
Cancel
Save