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

ClipByNormNoDivSum

产品支持情况

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

功能说明

  • 算子功能:执行四输入元素的ClipByNorm变体计算,最后不做除以norm的归一化,而是通过Select/Max链实现分段映射。公式描述如下:

  • 计算公式:

yi=max⁡(select_ones(xi≤greater_zerosi,xi,sqrt(select_ones(xi>greater_zerosi,xi,select_onesi))),maximum_onesi)y_i = \max(\text{select\_ones}(x_i \leq \text{greater\_zeros}_i, x_i, \text{sqrt}(\text{select\_ones}(x_i > \text{greater\_zeros}_i, x_i, \text{select\_ones}_i))), \text{maximum\_ones}_i)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 第一个输入tensor,公式中的x。 FLOAT、FLOAT16 ND
greater_zeros 输入 第二个输入tensor,比较阈值,公式中的greater_zeros。 FLOAT、FLOAT16 ND
select_ones 输入 第三个输入tensor,条件选择备用值,公式中的select_ones。 FLOAT、FLOAT16 ND
maximum_ones 输入 第四个输入tensor,最大值裁剪边界,公式中的maximum_ones。 FLOAT、FLOAT16 ND
y 输出 计算结果tensor,公式中的y。 FLOAT、FLOAT16 ND

约束说明

  • 所有输入输出的数据类型必须一致。
  • 所有输入支持Broadcast语义,输出shape为四个输入broadcast后的shape。

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_clip_by_norm_no_div_sum 通过算子IR构图方式调用ClipByNormNoDivSum算子。