文件最后提交记录最后更新时间
14 天前
1 个月前
1 个月前
1 个月前
1 个月前
1 个月前
3 个月前
30 天前
README

Swish

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
skywang2 个人开发者 Swish 2025/12/31 新增Swish算子

支持的产品型号

  • Atlas A2训练系列产品

产品形态详细说明请参见昇腾产品形态说明

算子描述

  • 功能描述

    Swish算子实现Swish激活函数,是一种由输入与其经过Sigmoid函数结果相乘得到的平滑、非线性函数。

  • 原型信息

    算子类型(OpType)Swish
    nameTypedata typeformat
    算子输入
    xtensorfloat32,float16,bfloat16ND
    算子输出 ytensorfloat32,float16,bfloat16ND
    算子属性 scalescalarfloat-
    核函数名swish

约束与限制

  • x,y的数据类型仅支持float32,float16,bfloat16,数据格式仅支持ND

运行验证

编译

bash build.sh --pkg --soc=ascend910b --experimental --ops=swish

运行

bash build.sh --run_example swish eager cust --vendor_name=custom --experimental
目录描述
test_aclnn_swish.cpp通过aclnn调用的方式调用Swish算子。