mindspore.ops.bernoulli

mindspore.ops.bernoulli(input, p=0.5, seed=None)

根据伯努利分布生成随机数（0或1）。

$$ou{t}_i\sim Bernoulli(p_i)$$

参数：

• input (Tensor) - 输入tensor。

• p (Union[Tensor, float], 可选) - 输出tensor中对应位置为1的概率。数值范围在0到1之间。默认 0.5 。

• seed (Union[int, None], 可选) - 随机种子。默认 None 表示使用时间戳。

返回：

Tensor

支持平台：

GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.tensor([1, 2, 3])
>>> mindspore.ops.bernoulli(input, p=1.0)
Tensor(shape=[3], dtype=Int64, value= [1, 1, 1])
>>> p = mindspore.tensor([0.0, 1.0, 1.0])
>>> mindspore.ops.bernoulli(input, p)
Tensor(shape=[3], dtype=Int64, value= [0, 1, 1])