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

Softshrink

产品支持情况

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

功能说明

  • 算子功能:对输入Tensor逐元素执行Softshrink激活函数。

  • 计算公式:

yi={xi−λ,if xi>λxi+λ,if xi<−λ0,otherwisey_i = \begin{cases} x_i - \lambda, & \text{if } x_i > \lambda \\ x_i + \lambda, & \text{if } x_i < -\lambda \\ 0, & \text{otherwise} \end{cases}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
input_x 输入 待进行Softshrink计算的输入Tensor。 fp16、fp32、bf16 ND
lambd 属性 Softshrink公式中的lambda值,默认0.5,需 >= 0。 float 标量
output_y 输出 Softshrink计算结果,shape与input_x相同。 fp16、fp32、bf16 ND

约束说明

  • lambd属性值必须 >= 0。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_softshrink.cpp 通过aclnn接口方式调用Softshrink算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
yourealize 个人开发者 Softshrink 2025/03 Softshrink算子适配开源仓