aclnnBernoulliTensor&aclnnInplaceBernoulliTensor

产品支持情况

[object Object]undefined

功能说明

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

• 计算公式：

  $$out∼Bernoulli(prob)$$

  其中，当使用aclnnBernoulliTensor时，公式中的prob对应第一段接口中的prob，公式中的out对应第一段接口中的out；当使用aclnnInplaceBernoulliTensor时，公式中的prob对应第一段接口中的prob，公式中的out对应第一段接口中的selfRef。

函数原型

• 每个算子分为，必须先调用“aclnnBernoulliTensorGetWorkspaceSize”或者“aclnnInplaceBernoulliTensorGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnBernoulliTensor”或者“aclnnInplaceBernoulliTensor”接口执行计算。

aclnnBernoulliTensorGetWorkspaceSize

• 参数说明：

[object Object]

• [object Object]Atlas 训练系列产品[object Object]：数据类型不支持UINT8、INT8、INT16、INT32、INT64、BOOL、BFLOAT16。

• [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：数据类型不支持UINT8、INT8、INT16、INT32、INT64、BOOL。

• 返回值：

aclnnStatus：返回状态码，具体参见

第一段接口完成入参校验，出现以下场景时报错：

[object Object]

aclnnBernoulliTensor

• 参数说明：

[object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

aclnnInplaceBernoulliTensorGetWorkspaceSize

• 参数说明：

[object Object]

• [object Object]Atlas 训练系列产品[object Object]：数据类型不支持BFLOAT16。

  • 返回值：

  aclnnStatus：返回状态码，具体参见

  第一段接口完成入参校验，出现以下场景时报错：

  [object Object]

aclnnInplaceBernoulliTensor

• 参数说明：

[object Object]

• 返回值：

  aclnnStatus: 返回状态码，具体参见。

约束说明

• 输入prob的元素值域必须为[0, 1]。
• 当输入prob的shape与输入self/selfRef的shape不一致时，只计算两者可对应元素的数据，其余数据的行为未定义。例如：当prob的shape为[4, 2]，self的shape为[4, 4]时，只计算前8个元素，输出的shape为[4, 4]；当prob的shape为[4, 4, 2], self的shape为[4, 4]时，只计算前16个元素，输出的shape为[4, 4]。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。