stateless_bernoulli

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas A2 推理系列产品	×
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

• 算子功能： 从伯努利分布中提取二进制随机数（0 或 1），prob为生成二进制随机数的概率，输入的张量用于指定shape。

• 计算公式：

  $$out\sim Bernoulli(prob)$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
shape	输入	随机数的shape。	INT32、INT64	ND
prob	输入	保活系数。	FLOAT16、BF16、FLOAT	ND
seed	输入	获取随机种子。	INT64	ND
offset	输入	获取值的步长。	INT64	ND
y	输出	获取输出的tensor。	INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL、FLOAT16、BFLOAT16、FLOAT	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn调用	test_aclnn_bernoulli_tensor	通过aclnnBernoulliTensor接口方式调用stateless_bernoulli算子。
aclnn调用	test_aclnn_bernoulli	通过aclnnBernoulli接口方式调用stateless_bernoulli算子。