Softshrink
产品支持情况
| 产品 |
是否支持 |
| Atlas A2 训练系列产品/Atlas 800I A2 推理产品 |
√ |
功能说明
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}
yi=⎩⎪⎪⎨⎪⎪⎧xi−λ,xi+λ,0,if xi>λif xi<−λotherwise
参数说明
| 参数名 |
输入/输出/属性 |
描述 |
数据类型 |
数据格式 |
| 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 |
约束说明
调用说明
贡献说明
| 贡献者 |
贡献方 |
贡献算子 |
贡献时间 |
贡献内容 |
| yourealize |
个人开发者 |
Softshrink |
2025/03 |
Softshrink算子适配开源仓 |