文件最后提交记录最后更新时间
7 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
8 个月前
README

DropOutV3

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT

功能说明

  • 算子功能:训练过程中,按照概率p随机将输入中的元素置零,并将输出按照1/(1-p)的比例缩放。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
input 输入张量 输入元素。 FLOAT、FLOAT16、BFLOAT16 ND
noise_shape 输入张量 预留参数,入参请用空指针代替。 INT64、INT32 ND
p 输入张量 元素置零的概率,取值范围为[0, 1] FLOAT、FLOAT16、BFLOAT16 ND
seed 输入张量 随机数的种子,影响生成的随机数序列 INT32、INT64 ND
offset 输入张量 随机数的偏移量,它影响生成的随机数序列的位置 INT64 ND
y 输出 输出数据。 同 input ND
mask 输出 bit类型并使用UINT8类型存储的mask数据。 UINT8 ND

约束说明

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_drop_out_v3 通过aclnn_drop_out_v3接口方式调用drop_out_v3算子。