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

Shrink

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品 ×

功能说明

  • 算子功能:对输入张量进行非线性收缩变换,根据输入值与阈值lambd的关系将元素映射到三个区域(正区域、负区域、死区),常用于稀疏化模型推理和训练中的权重正则化。

  • 计算公式

    outi={selfi−bias,selfi>lambdselfi+bias,selfi<−lambd0,−lambd≤selfi≤lambdout_i=\begin{cases} self_i-bias, & self_i > lambd \\ self_i+bias, & self_i < -lambd \\ 0, & -lambd \leq self_i \leq lambd \end{cases}

    其中lambd为阈值参数(默认0.5),bias为偏移量参数(默认0.0)。当lambd < 0时,内部钳位为0。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
input_x 输入 输入张量,对应公式中的self。元素值与lambd比较决定归属区域。 FLOAT16、FLOAT ND
lambd 属性 阈值参数,对应公式中的lambd。当self_i>lambd时进入正区域,self_i<-lambd时进入负区域。默认值为0.5。当lambd<0时,内部钳位为0。 FLOAT -
bias 属性 偏移量参数,对应公式中的bias。在正区域减去bias,在负区域加上bias。默认值为0.0。 FLOAT -
output_y 输出 输出张量,对应公式中的out。与input_x的dtype和shape完全相同。 FLOAT16、FLOAT ND

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_shrink 通过aclnnShrink接口方式调用Shrink算子。
图模式 - 通过算子IR构图方式调用Shrink算子。