bernoulli¶

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

该OP以输入 x 为概率，生成一个伯努利分布（0-1分布）的Tensor，输出Tensor的形状和数据类型与输入 x 相同。

\[out_i \sim Bernoulli(p = x_i)\]

参数¶

x (Tensor) - 输入的概率值。数据类型为 float32 、float64 .

name (str, 可选) - 操作的名称(可选，默认值为None）。更多信息请参见 Name。

返回¶

Tensor：伯努利分布的随机Tensor，形状和数据类型为与输入 x 相同。

代码示例¶

          import paddle

paddle.seed(100) # on CPU device
x = paddle.rand([2,3])
print(x)
# [[0.5535528  0.20714243 0.01162981]
#  [0.51577556 0.36369765 0.2609165 ]]

out = paddle.bernoulli(x)
print(out)
# [[0. 0. 0.]
#  [1. 1. 0.]]

         