bernoulli

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

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

bernoulli - 图1

参数:

  • x (Tensor) - 输入的概率值。数据类型为 float32float64 .

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

返回:

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

代码示例

  1. import paddle
  2. paddle.seed(100) # on CPU device
  3. x = paddle.rand([2,3])
  4. print(x)
  5. # [[0.5535528 0.20714243 0.01162981]
  6. # [0.51577556 0.36369765 0.2609165 ]]
  7. out = paddle.bernoulli(x)
  8. print(out)
  9. # [[0. 0. 0.]
  10. # [1. 1. 0.]]