binomial¶

paddle. binomial ( count, prob, name=None ) [源代码] ¶

以输入参数 count 和 prob 分别为二项分布的 n 和 p 参数，生成一个二项分布的随机数 Tensor ，支持 Tensor 形状广播。输出 Tensor 的 dtype 为 int64 。

\[out_i \sim Binomial (n = count_i, p = prob_i)\]

参数¶

count (Tensor) - Tensor 的每个元素代表一个二项分布的总试验次数。数据类型支持 int32 、int64 。

prob (Tensor) - Tensor 的每个元素代表一个二项分布的试验成功概率。数据类型支持 bfloat16 、float16 、float32 、float64 。

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

返回¶

Tensor，二项分布采样得到的随机 Tensor，形状为 count 和 prob 进行广播后的 Tensor 形状， dtype 为 int64 。

代码示例¶

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

>>> n = paddle.to_tensor([10.0, 50.0])
>>> p = paddle.to_tensor([0.2, 0.6])
>>> out = paddle.binomial(n, p)
>>> print(out)
Tensor(shape=[2], dtype=int64, place=Place(cpu), stop_gradient=True,
[1 , 31])

         