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

HardSwish

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

  • 算子功能:对输入张量x逐元素计算HardSwish激活函数值,计算公式为: y=x∗min(max(x+3,0),6)6y = \frac{x * min(max(x + 3, 0), 6)}{6}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
self 输入 x float、half、bfloat16 ND
out 输出 y float、half、bfloat16 ND

约束说明

待更新

目前支持float32、float16、bfloat16数据类型,不支持broadcast,后续会持续改进

调用说明

调用方式 样例代码 说明
aclnn接口 test_hard_swish_v2 通过[aclnnHardSwishV2]接口方式调用HardSwish算子。