forked from huawei/mindspore2022
212 lines
5.4 KiB
ReStructuredText
212 lines
5.4 KiB
ReStructuredText
mindspore.nn.probability.distribution.Geometric
|
||
================================================
|
||
|
||
.. py:class:: mindspore.nn.probability.distribution.Geometric(probs=None, seed=None, dtype=mstype.int32, name='Geometric')
|
||
|
||
几何分布（Geometric Distribution）。
|
||
|
||
它代表在第一次成功之前有k次失败，即在第一次成功实现时，总共有k+1个伯努利试验。
|
||
离散随机分布，取值范围为正自然数集，概率质量函数为 :math:`P(X = i) = p(1-p)^{i-1}, i = 1, 2, ...`。
|
||
|
||
**参数：**
|
||
|
||
- **probs** (int, float, list, numpy.ndarray, Tensor) - 成功的概率。默认值：None。
|
||
- **seed** (int) - 采样时使用的种子。如果为None，则使用全局种子。默认值：None。
|
||
- **dtype** (mindspore.dtype) - 事件样例的类型。默认值：mindspore.int32.
|
||
- **name** (str) - 分布的名称。默认值：'Geometric'。
|
||
|
||
.. note::
|
||
`probs` 必须是合适的概率（0<p<1）。
|
||
|
||
|
||
**异常：**
|
||
|
||
- **ValueError** - `probs` 中元素小于0或者大于1。
|
||
|
||
.. py:method:: probs
|
||
:property:
|
||
|
||
返回伯努利试验成功的概率。
|
||
|
||
**返回：**
|
||
|
||
Tensor，伯努利试验成功的概率值。
|
||
|
||
.. py:method:: cdf(value, probs)
|
||
|
||
在给定值下计算累积分布函数（Cumulatuve Distribution Function, CDF）。
|
||
|
||
**参数：**
|
||
|
||
- **value** (Tensor) - 要计算的值。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，累积分布函数的值。
|
||
|
||
.. py:method:: cross_entropy(dist, probs_b, probs)
|
||
|
||
计算分布a和b之间的交叉熵。
|
||
|
||
**参数：**
|
||
|
||
- **dist** (str) - 分布的类型。
|
||
- **probs_b** (Tensor) - 对比分布的伯努利实验成功的概率。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，交叉熵的值。
|
||
|
||
.. py:method:: entropy(probs)
|
||
|
||
计算熵。
|
||
|
||
**参数：**
|
||
|
||
- **probs_b** (Tensor) - 对比分布的伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，熵的值。
|
||
|
||
.. py:method:: kl_loss(dist, probs)
|
||
|
||
计算KL散度，即KL(a||b)。
|
||
|
||
**参数：**
|
||
|
||
- **dist** (str) - 分布的类型。
|
||
- **probs_b** (Tensor) - 对比分布的伯努利实验成功的概率。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，KL散度。
|
||
|
||
.. py:method:: log_cdf(value, probs)
|
||
|
||
计算给定值对于的累积分布函数的对数。
|
||
|
||
**参数：**
|
||
|
||
- **value** (Tensor) - 要计算的值。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，累积分布函数的对数。
|
||
|
||
.. py:method:: log_prob(value, probs)
|
||
|
||
计算给定值对应的概率的对数。
|
||
|
||
**参数：**
|
||
|
||
- **value** (Tensor) - 要计算的值。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，累积分布函数的对数。
|
||
|
||
.. py:method:: log_survival(value, probs)
|
||
|
||
计算给定值对应的生存函数的对数。
|
||
|
||
**参数：**
|
||
|
||
- **value** (Tensor) - 要计算的值。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，生存函数的对数。
|
||
|
||
.. py:method:: mean(probs)
|
||
|
||
计算期望。
|
||
|
||
**参数：**
|
||
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，概率分布的期望。
|
||
|
||
.. py:method:: mode(probs)
|
||
|
||
计算众数。
|
||
|
||
**参数：**
|
||
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，概率分布的众数。
|
||
|
||
.. py:method:: prob(value, probs)
|
||
|
||
计算给定值下的概率。对于离散分布是计算概率质量函数（Probability Mass Function）。
|
||
|
||
**参数：**
|
||
|
||
- **value** (Tensor) - 要计算的值。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，概率值。
|
||
|
||
.. py:method:: sample(shape, probs)
|
||
|
||
采样函数。
|
||
|
||
**参数：**
|
||
|
||
- **shape** (tuple) - 样本的shape。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，根据概率分布采样的样本。
|
||
|
||
.. py:method:: sd(probs)
|
||
|
||
计算标准差。
|
||
|
||
**参数：**
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，概率分布的标准差。
|
||
|
||
.. py:method:: survival_function(value, probs)
|
||
|
||
计算给定值对应的生存函数。
|
||
|
||
**参数：**
|
||
|
||
- **value** (Tensor) - 要计算的值。
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，生存函数的值。
|
||
|
||
.. py:method:: var(probs)
|
||
|
||
计算方差。
|
||
|
||
**参数：**
|
||
|
||
- **probs** (Tensor) - 伯努利实验成功的概率。默认值：None。
|
||
|
||
**返回：**
|
||
|
||
Tensor，概率分布的方差。
|