poisson

paddle. poisson ( x, name=None ) [源代码]

以输入参数 x 为泊松分布的 lambda 参数,生成一个泊松分布的随机数 Tensor,输出 Tensor 的 shape 和 dtype 与输入 Tensor 相同。

\[out_i \sim Poisson(lambda = x_i)\]

参数

  • x (Tensor) - Tensor 的每个元素,对应泊松分布的 lambda 参数。数据类型为:bfloat16, float16, float32、float64。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

Tensor:泊松分布的随机数 Tensor,shape 和 dtype 与输入 x 相同。

代码示例

>>> import paddle
>>> paddle.set_device('cpu')
>>> paddle.seed(100)

>>> x = paddle.uniform([2,3], min=1.0, max=5.0)
>>> out = paddle.poisson(x)
>>> print(out)
Tensor(shape=[2, 3], dtype=float32, place=Place(cpu), stop_gradient=True,
[[2., 5., 0.],
 [5., 1., 3.]])