DropOutDoMaskV3D

产品支持情况

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

功能说明

• 算子功能：训练过程中，按照概率keep_prob随机保留输入中的元素，并将输出按照1/keep_prob的比例放大。若mask对应比特位为1，则输入中相应的元素放大。若mask中比特位为0，则gradOutput相应的元素置零。特别地，若keep_prob为0，则将所有元素置为0；若keep_prob为1，则不改变gradOutput的元素。

• 计算公式：

$$gradOutpu{t}_i=\{\begin{array}{ll}0,& mas{k}_i=0\\ \frac{1}{keep\_prob}sel{f}_i,& mas{k}_i=1\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入	公式中的输入self，shape支持0-8维。	FLOAT、FLOAT16、BFLOAT16	ND
mask	输入	bit类型并使用UINT8类型存储的mask数据，shape需要为(align(self的元素个数,128)/8)。	UINT8、BOOL	ND
keep_prob	属性	用于计算输出数据缩放比例的概率值。	FLOAT	-
y	输出	公式中的gradOutput，数据类型需要是self可转换的数据类型，shape需要与self一致。	FLOAT、FLOAT16、BFLOAT16	ND

约束说明

1. keep_prob的值必须在0和1之间。
2. mask的shape必须满足条件：$$\text{mask_shape} = \frac{\text{align}(\text{num}(self), 128)}{8}$$
3. 数据维度支持0-8维。

调用说明

调用方式	样例代码	说明
图模式调用	test_geir_drop_out_do_mask_v3_d	通过算子IR构图方式调用TruncatedNormalV2算子。