Renorm

产品支持情况

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

功能说明

• 算子功能：返回一个张量，其中输入张量self沿维度dim的每个子张量都经过归一化，使得子张量的p范数低于maxNorm值。

• 计算公式：

  $$outpu{t}_i=\{\begin{array}{r}inpu{t}_i,||inpu{t}_i|{|}_p<=maxNorm\\ \frac{inpu{t}_i}{||inpu{t}_i|{|}_p}\cdot maxNorm,||inpu{t}_i|{|}_p>maxNorm\end{array}$$

  其中： $i$为dim确定的某维度张量切片：

  $$||inpu{t}_i|{|}_p=(\sum _{i=0}^n{inpu{t}_i^p}^{\frac{1}{p}})$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	表示进行重归一化计算的输入，对应公式中的`input`。	FLOAT32、FLOAT16、BFLOAT16	ND
p	属性	表示范数，对应公式中的`p`。取值大于等于0。	FLOAT32	-
dim	属性	表示指定求norm的维度方向。对应公式中的`i`。取值范围为：[-x的维度数量，x的维度数量-1]。	INT64	-
maxNorm	属性	表示最大允许的归一化值。对应公式中的`maxNorm`。取值大于等于0。如果运算时对应维度的`p`范数（由`p`值确定）大于`maxNorm`，则将该维度的值关于`p`范数归一化并乘上`maxNorm`。如果运算时对应维度的`p`范数（由`p`值确定）小于`maxNorm`，则该维度张量保持不变输出。	FLOAT32	-
y	输出	表示最终输出，对应公式中的`output`。数据类型、shape与入参`x`保持一致。	FLOAT32、FLOAT16、BFLOAT16	ND

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_renorm	通过aclnnRenorm接口方式调用Renorm算子。
aclnn接口	test_aclnn_inplace_renorm	通过aclnnInplaceRenorm接口方式调用Renorm算子。