mindspore.ops.Bernoulli

class mindspore.ops.Bernoulli(seed=- 1, offset=0)[源代码]

以 p 的概率随机将输出的元素设置为0或1，服从伯努利分布。

警告

这是一个实验性API，后续可能修改或删除。

更多细节请参考 mindspore.ops.bernoulli() 。

参数：

seed (int, 可选) - 随机种子，用于生成随机数，数值范围是-1或正整数，-1代表取当前时间戳。默认值：-1。
offset (int, 可选) - 用于在生成随机数序列时改变起始位置。默认值：0。

输入：

x (Tensor) - Tensor的输入，其数据类型为int8、uint8、int16、int32、int64、bool、float32或float64。
p (Union[Tensor, float], 可选) - 成功概率。 p 中每个值代表输出Tensor中对应位置为1的概率，如果是Tensor，其shape必须与 input 一致，数值范围在0到1之间。默认值：0.5。

输出：

y (Tensor) - shape和数据类型与 x 相同。

支持平台：

GPU CPU

样例：

>>> input_x = Tensor([0.1, 0.2, 0.3], mindspore.float32)
>>> bernoulli = ops.Bernoulli()
>>> output = bernoulli(input_x, Tensor([1.0]))
>>> print(output)
[1. 1. 1.]
>>> input_p = Tensor([0.0, 1.0, 1.0], mindspore.float32)
>>> output = bernoulli(input_x, input_p)
>>> print(output)
[0. 1. 1.]