mindspore.nn.probability.distribution.TransformedDistribution

查看源文件
class mindspore.nn.probability.distribution.TransformedDistribution(bijector, distribution, seed=None, name='transformed_distribution')[源代码]

转换分布(Transformed Distribution)。 该类包含一个Bijector和一个分布,并通过Bijector定义的操作将原始分布转换为新分布。可如果原始分布为 \(X\) ,Bijector的映射函数为 \(g(x)\),那么对应的转换分布为 \(Y = g(X)\)

参数:
  • bijector (Bijector) - 要执行的转换。

  • distribution (Distribution) - 原始分布。必须具有float数据类型。

  • seed (int) - 采样时使用的种子。如果为None,则使用全局种子。默认值: None 。如果在初始化TransformedDistribution对象时给出了此种子,则对象的采样函数将使用此种子;否则,将使用基础分布的种子。

  • name (str) - 转换分布的名称。默认值: 'transformed_distribution'

说明

用于初始化原始分布的参数不能为None。例如,由于未指定 meansd ,因此无法使用mynormal = msd.Normal(dtype=mindspore.float32)初始化TransformedDistribution。 batch_shape 为原始分布的 batch 的 shape。 broadcast_shape 为原始分布和概率映射经过广播后的 shape。 is_scalar_batch 为 True 当且仅当原始分布和概率映射的 batch 均为标量。 default_parametersparameter_namesparameter_type 由原始分布的数据类型决定。 衍生类可以通过调用 reset_parameters 后再调用 add_parameter 来添加参数以覆盖 default_parametersparameter_names

异常:
  • TypeError - bijector不是Bijector类。

  • TypeError - distribution不是Distribution类。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> import mindspore
>>> import mindspore.nn as nn
>>> import mindspore.nn.probability.distribution as msd
>>> import mindspore.nn.probability.bijector as msb
>>> from mindspore import Tensor
>>> class Net(nn.Cell):
...     def __init__(self, shape, dtype=mindspore.float32, seed=0, name='transformed_distribution'):
...         super(Net, self).__init__()
...         # create TransformedDistribution distribution
...         self.exp = msb.Exp()
...         self.normal = msd.Normal(0.0, 1.0, dtype=dtype)
...         self.lognormal = msd.TransformedDistribution(self.exp, self.normal, seed=seed, name=name)
...         self.shape = shape
...
...     def construct(self, value):
...         cdf = self.lognormal.cdf(value)
...         sample = self.lognormal.sample(self.shape)
...         return cdf, sample
>>> shape = (2, 3)
>>> net = Net(shape=shape, name="LogNormal")
>>> x = np.array([2.0, 3.0, 4.0, 5.0]).astype(np.float32)
>>> tx = Tensor(x, dtype=mindspore.float32)
>>> cdf, sample = net(tx)
>>> print(sample.shape)
(2, 3)
property bijector

返回概率映射函数。

返回:

Bijector,概率映射函数。

property distribution

返回变化前的概率分布。

返回:

Distribution,变化前的概率分布。

property dtype

返回分布的数据类型。

返回:

mindspore.dtype,分布的数据类型。

property is_linear_transformation

返回概率映射函数是否为线性映射。

返回:

Bool,概率映射函数为线性映射则返回True,否则返回False。

cdf(value)

在给定值下计算累积分布函数(Cumulatuve Distribution Function, CDF)。

参数:
  • value (Tensor) - 要计算的值。

返回:

Tensor,累积分布函数的值。

log_cdf(value)

计算给定值对于的累积分布函数的对数。

参数:
  • value (Tensor) - 要计算的值。

返回:

Tensor,累积分布函数的对数。

log_prob(value)

计算给定值对应的概率的对数。

参数:
  • value (Tensor) - 要计算的值。

返回:

Tensor,累积分布函数的对数。

log_survival(value)

计算给定值对应的生存函数的对数。

参数:
  • value (Tensor) - 要计算的值。

返回:

Tensor,生存函数的对数。

mean()

计算期望。

返回:

Tensor,概率分布的期望。

prob(value)

计算给定值下的概率。

参数:
  • value (Tensor) - 要计算的值。

返回:

Tensor,概率值。

sample(shape)

采样函数。

参数:
  • shape (tuple) - 样本的shape。

返回:

Tensor,根据概率分布采样的样本。

survival_function(value)

计算给定值对应的生存函数。

参数:
  • value (Tensor) - 要计算的值。

返回:

Tensor,生存函数的值。