mindspore.ops.Poisson

class mindspore.ops.Poisson(seed=0, seed2=0)

生成 Poisson 分布的随机数。

\[\text{P}(i|μ) = \frac{\exp(-μ)μ^{i}}{i!},\]

参数：

• seed (int) - 算子层的随机种子，用于生成随机数。必须是非负的。默认值：0。

• seed2 (int) - 全局的随机种子，和算子层的随机种子共同决定最终生成的随机数。必须是非负的。默认值：0。

输入：

• shape (tuple) - 待生成的随机 Tensor 的 shape。只支持常量值。

• mean (Tensor) - Poisson 分布的期望，也就是上面公式中的 μ。其值必须大于0。数据类型为 float32。

输出：

Tensor。shape是输入 shape 和 mean 广播后的 shape。数据类型为 int32。

异常：

• TypeError - seed 或 seed2 的数据类型不是 int。

• TypeError - shape 不是 tuple。

• TypeError - mean 不是数据类型为 float32 的 Tensor。

支持平台：

Ascend

样例：

>>> shape = (4, 1)
>>> mean = Tensor(np.array([5.0, 10.0]), mstype.float32)
>>> poisson = ops.Poisson(seed=5)
>>> output = poisson(shape, mean)
>>> result = output.shape
>>> print(result)
(4, 2)